[PYTHON/TENSORFLOW] 파이썬 텐서플로 GPU 버전 설치하기

1. 엔비디아 사이트(https://developer.nvidia.com/cuda-toolkit-archive)에서 Cuda 9.0 버전을 다운로드 받아 설치한다.

• cuda_9.0.176_win10.exe
• cuda_9.0.176.1_windows.exe
• cuda_9.0.176.2_windows.exe
• cuda_9.0.176.3_windows.exe
• cuda_9.0.176.4_windows.exe

2. 엔비디아 사이트(https://developer.nvidia.com/cudnn)에서 CuDNN 7.5.0 버전을 다운로드 한다.

cudnn-9.0-windows10-x64-v7.5.0.56.zip

 

728x90

 

3. 해당 압축 파일을 압축 해제하고 아래와 같이 압축 해제 폴더의 파일을 타겟 폴더에 복사한다.

압축 해제 폴더	타겟 폴더
cuda\bin	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
cuda\include	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
cuda\lib	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib

 

300x250

 

4. 파이썬 사이트(https://www.python.org/downloads/windows)에서 파이썬 3.6.8 버전을 다운로드 받아 설치한다.

python-3.6.8-amd64.exe

 

5. 명령 프롬프트를 관리자 모드로 실행한다.

 

6. 아래 명령을 실행해 텐서플로 1.6 버전을 설치한다.

pip3 install --upgrade tensorflow-gpu==1.6