Windows 10 下載CUDA9.2 + cuDNNv7.2.1 + tensorflow-gpu 1.9.0
CUDA安裝
根據官網描述在下載CUDA之前請確保:
- GPU型號相容
- 支援的Windows版本
- 支援的Microsoft Visual Studio版本
cuda下載地址:http://developer.nvidia.com/cuda-downloads
這裡我安裝的是Microsoft Visual Studio 2017,但是之前在安裝CUDA 9.2時總是安裝失敗,於是又嘗試了9.1 ,9.0都是一樣的問題,安裝失敗,所以我判斷是Microsoft Visual Studio 的問題。之後重灌了Microsoft Visual Studio 2017
輸入nvcc --version或者nvcc -V驗證:
cuDNN下載
需要註冊一個官網帳號, 下載地址:https://developer.nvidia.com/rdp/cudnn-download
解壓縮下載的CuDnn檔案,得到3個資料夾:bin, include, lib。
將這個三個資料夾複製到“C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.2” (CUDA安裝路徑)
之後確認環境變數:
確認CUDA_PATH和CUDA_PATH_V8.0已經存在。
手動新增 “C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.2\bin”到Path裡面。
pip安裝Tensorflow-gpu
注意:
目前tensorflow還不能支援cuda9.2,所以只能通過原始碼編譯或者別人編譯好的安裝包安裝。大神編譯的各種版本的tensorflow安裝包地址:https://github.com/fo40225/tensorflow-windows-wheel
根據說明找到資源路徑:
下載完成後:
將tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl
複製到C:\Windows\System32資料夾下,然後執行命令:pip install tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl
最後pip list檢視:
執行測試
完工~~體驗gpu的運算速度吧