■ 파이썬 텐서플로 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

 

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

 

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

 

------------------------------------------------------------------------------------------------------------------------

Posted by 사용자 icodebroker

댓글을 달아 주세요