1. 程式人生 > >Windows 10 下載CUDA9.2 + cuDNNv7.2.1 + tensorflow-gpu 1.9.0

Windows 10 下載CUDA9.2 + cuDNNv7.2.1 + tensorflow-gpu 1.9.0

CUDA安裝

根據官網描述在下載CUDA之前請確保:

  1. GPU型號相容
  2. 支援的Windows版本
  3. 支援的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

(若執行失敗,會報錯在哪個資料夾不能找到該檔案,再把.whl檔案移到指定資料夾即可)

最後pip list檢視:

執行測試

完工~~體驗gpu的運算速度吧