1. 程式人生 > >win10下,配置tensorflow+GPU

win10下,配置tensorflow+GPU

首先,需要三個軟體包,一個Anaconda,一個CUDA,一個CUDNN

Anaconda網上下載相對應的系統版本就行了

選擇下載CUDA8.0,則應該選擇Cudnn6.0!!!!!!如果選擇CUDA9.0,CUDNN選擇相對應的Cudnn(推薦前者)

下面關於CUDA部分是網上參考的內容,用的是cuda9.0,安裝8.0的時候新增環境變數這些其實是一樣的,只是要注意最後一定要選擇不同的CUDA所對應的CUDNN,,而且cuda8.0,安裝tensorflow的時候是需要指定版本的。

一、安裝順序

1、首先安裝CUDA9.0/8.0。下載地址。

選擇9.0版本是因為網上說,tensorflow的windows-gpu還不支援CUDA的新版本。再加上同學安裝的也是9.0,用的沒有問題,所以就選擇了這個版本。

選擇8.0版本,安裝tensorflow gpu時就要選擇相對應的1.4.0版本

在下載的時候選擇適合自己的版本。     

我選擇的如上圖,剛開始第四行選擇的是右邊的那個,但是下載下來安裝不了,百度查是電腦缺少什麼壓縮軟體,所以建議選左邊的那個。

然後只需要選擇下圖的base installer即可。

開啟下載好的CUDA9.0/8.0開始安裝,剛開始的安裝程式存放位置,自己隨意就好。

下一步會檢測系統相容性,有些顯示卡是不支援GPU的,自己需要先查清楚。

下一步接受協議

下一步選擇安裝模式,我選擇的是自定義,程式預設的精簡模式應該可以理解為安裝所有東西,其中包括了我暫時不用的VS以及顯示卡驅動,所以我選擇的是自定義模式。這也是查了其他的博文所做出的選擇。

下一步的選擇為

下一步會讓選擇安裝路徑,我最初是自己設定了路徑,因為不想往C盤放東西,可是安裝完之後,在我設定的路徑裡沒找到安裝的CUDA,他還是安裝在了C盤,並且沒有安裝完全。所以我安裝了第二次,我並沒有解除安裝第一次裝的,因為沒找到解除安裝的方法。直接開始了第二次安裝,還是選擇自己設定的路徑,還是沒安裝成功。很難受,所以第三次安裝直接選擇他的預設路徑,一點多個G還能接受,放在C盤也問題不大。這次安裝成功了。

所以個人建議,如果C盤能接受一個多G,不如直接安裝在預設路徑,防止老出問題。

安裝完成之後,下一步需要配置環境變數。

如果安裝成功,系統變數會多出兩個為

還需要自己新增幾個系統變數

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0(這是預設安裝位置的路徑,如果自己路徑設定安裝成功的話就用自己的路徑)

CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 

CUDA_BIN_PATH = %CUDA_PATH%\bin 

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

新增好之後應該像下面這樣。

下一步在系統變數PATH裡新增東西。找到系統變數的PATH雙擊。

新增下列東西。

%CUDA_LIB_PATH%

%CUDA_BIN_PATH%

%CUDA_SDK_LIB_PATH%

%CUDA_SDK_BIN_PATH%

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64   這些均為預設路徑,有需要的話自行修改

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64

新增完之後CUDA就算安裝完成了。我們可以檢驗是否安裝成功。

開啟CMD,cd到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite    (這是預設路徑)

分別執行命令:bandwidthTest.exe

                         deviceQuery.exe

若分別返回

 

則代表CUDA9.0/8.0安裝成功。

2、安裝cudnn。

官網上下載搭配CUDA9.0的cudnn ,一定要注意搭配9.0。會需要註冊英偉達的賬號。

如果想使用下載cuda8.0,要注意搭配8.0的cudnn(我用的cudnn6.0來搭配cuda8.0)

下載好之後,解壓,分別將.h .lib 和.dll 檔案分別拷貝到cuda的include, lib/x64, bin 資料夾下。其實也就是對應的資料夾。

到這裡就安裝好了CUDA9.0/8.0及其對應的cudnn。

3.然後就把下好的anaconda包進行安裝,安裝到最後一步時記得有提醒是否加入環境變數,勾上就行了

4.然後進入cmd,pip install tensorflow-gpu就行了

 !!!!!!如果你選擇的是cuda8.0的話,這個地方記得 pip install tensorflow-gpu=1.4.0

然後就完全配置好了