wilson's story

윈도우 NVIDIA CUDA 설치하기 본문

Python

윈도우 NVIDIA CUDA 설치하기

wilson 2019. 7. 29. 14:52
반응형

CUDA Toolkit 과 cuDNN SDK 설치

https://www.tensorflow.org/install/gpu
에서 보면 소프트웨어 요구사항 항목에서 

CUDA Toolkit 다운로드 경로
https://developer.nvidia.com/cuda-toolkit-archive

cuDNN SDK 다운경로 경로
회원가입후 사용가능
https://developer.nvidia.com/rdp/cudnn-download


CUDA Toolkit 와 cuDNN SDK 는 버전에 맞게 다운 받아준다.
예) 
CUDA Toolkit 10.1
Download cuDNN v7.6.2 (July 22, 2019), for CUDA 10.1

설치순서

1. CUDA Toolkit 설치
설치위치 기억하고 있음
예) 
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

2. cuDNN SDK 압축을 풀면 
bin, include, lib 폴더를 있음



CUDA Toolkit 설치 위치에다가 덮어씌움


확인방법 
anaconda env 생성 

conda create -n env이름 python=3.7

activate env이름

pip install tensorflow-gpu


import tensorflow as tf
a=tf.constant(3)
sess = tf.Session()
으로 확인







반응형