win7/win10下cuda8.0安裝跑gpu版tensorflow
簡單寫下配置過程吧,挺簡單的:
0.win7/10 X64系統
1.安裝vs2013
2.安裝Anaconda3(需要裡面的python環境)
3.下載CUDA8.0(https://developer.nvidia.com/cuda-downloads),下面兩個exe檔案都下載
補充說明:我電腦裡原來是cuda7.5,故需要先解除安裝乾淨,我是這麼做的,有需要的可以借鑑:
(1)把下圖中的幾項用電腦管家全部解除安裝乾淨
(2)刪除C:\Program Files\NVIDIA GPU Computing Toolkit 資料夾
刪除 C:\ProgramData\NVIDIA GPU Computing Toolkit 資料夾
刪除C:\ProgramData\NVIDIA Corporation\CUDA Samples 資料夾
4.執行exe
預設下一步到最後
5.再執行exe
預設下一步到最後
6.下載cuddn5.1/cudnn6.0(如果想要了解cuddn和cuda的區別可以看該部落格http://blog.csdn.net/fangjin_kl/article/details/53906874
7.解壓cuddn5.1/cudnn6.0,把如下的三個資料夾替換到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0夾下(該目錄是我的cuda8.0的安裝目錄)
這裡需要在系統變數裡設定下面幾個變數:
CUDA_PATH: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
CUDA_BIN_PATH: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
CUDA_LIB_PATH: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
CUDA_PATH_V8_0: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
CUDA_SDK_BIN_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64
CUDA_SDK_LIB_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64
CUDA_SDK_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
8.檢視cuda8.0是否安裝成功,可以在CMD視窗下敲指令nvcc -V
9.再執行一個Sample例子 開啟C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\1_Utilities\deviceQuery解決方案,編譯下,出來的結果為
10.pip安裝tensorflow gpu庫(http://blog.csdn.net/u014365862/article/details/53868578)
在Anaconda Prompt裡輸入(不是cmd跳出的視窗內輸入) pip install tensorflow-gpu
11.安裝完後輸入import tensorflow試試
其實win7,win10的安裝都一個樣,關鍵是要檔案下載對,然後還有一個注意的地方就是設定環境變數的時候不要弄錯了,預設cuda安裝的路徑,就直接複製貼上上面的路徑就好了。