1. 程式人生 > >win7/win10下cuda8.0安裝跑gpu版tensorflow

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安裝的路徑,就直接複製貼上上面的路徑就好了。