■ 파이썬 텐서플로 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. 해당 압축 파일을 압축 해제하고 아래와 같이 압축 해제 폴더의 파일을 타겟 폴더에 복사한다.
▶ 표
|
1 2 3 4 5 6 7 8 9 |
──────────────────────────────────────── 압축 해제 폴더 타겟 폴더 ─────── ──────────────────────────────── 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 버전을 설치한다.
▶ 실행 명령
|
1 2 3 |
pip3 install --upgrade tensorflow-gpu==1.6 |