win10下基於anaconda安裝tensorflow-gpu


1.最重要的一點就是,一定要知道你要安裝的tensorflow版本跟你的cuda以及cudnn版本是否匹配。小白本人在這里被坑了無數次,以至於一度懷疑人生,花費了我將近一天半的時間。

那么,該如何判斷呢?下面是小白找的表:

小白的anaconda對應的python是3.6.0,在這里附上本次安裝所要用到的資源鏈接:

 鏈接:https://pan.baidu.com/s/1Vn8cPzEPQQeB7emz4PefLA
提取碼:8a1m
復制這段內容后打開百度網盤手機App,操作更方便哦

 

2.cuda安裝:

下載好文件,先點擊cuda_8.0.61_win10那個大文件,安裝。

安裝過程如下,

當然點同意了,選擇自定義安裝,

點擊下一步,下一步,同意.....等待一小會。

 

接下來安裝cuda的補丁,就是剛剛下載下來的小文件,

安裝好過后,解壓cudnn-8.0-windows10-x64-v6.0

得到三個文件夾

然后按照圖片中路徑找到相應的文件夾點開,

把剛剛解壓出來的三個文件夾中的文件,注意注意,是文件,不是文件夾,分別放到c盤中安裝的cuda文件所對應的文件夾中

 

接下來在計算機圖標那里右鍵,屬性,高級系統設置,環境變量,

選擇Path雙擊打開,增加環境變量,

 

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

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

 

選擇上移把這兩個移到上面:

 

然后安裝anaconda:

anaconda就直接下一步下一步,小白嘗試過安裝在其它盤,都失敗了,所以就默認安裝就好了!

最好都勾上。

安裝完成后,點擊開始(菜單欄)Anaconda Prompt。創建環境:

conda create --name tensorflow-gpu python=3.6

輸入y,回車鍵

環境建立好后,輸入:activate tensorflow-gpu

再輸入:pip install tensorflow-gpu==1.4.0

耐心等待,如果出現timeout 再輸入pip install tensorflow-gpu==1.4.0

這個下載速度非常慢............

 

顯示成功安裝后,輸入python,

然后import tensorflow,沒有報錯說明安裝成功。

 

安裝成功后,安裝好pycharm,pycharm在我分享的文件中,安裝好后需要把剛剛建立的tensorflow-gpu的環境添加到pycharm中來,操作如下:

打開,file->setting->搜索:Project Interpreter

 

 

進入C盤:C:\Users\haibo\Anaconda3\envs\tensorflow-gpu\python.exe,添加環境。

 

之后創建一個pythonfile,運行

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

結果如下:
C:\Users\haibo\Anaconda3\envs\tensorflow-gpu\python.exe C:/Users/haibo/Desktop/age-gender-estimation-master/哈哈哈哈哈.py
2019-03-03 09:06:22.602887: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2019-03-03 09:06:23.718693: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\common_runtime\gpu\gpu_device.cc:1030] Found device 0 with properties:
name: GeForce GTX 1050 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.62
pciBusID: 0000:01:00.0
totalMemory: 4.00GiB freeMemory: 3.30GiB
2019-03-03 09:06:23.719159: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 6235084662034035804
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3225865420
locality {
  bus_id: 1
}
incarnation: 13303771124769821445
physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"
]
Process finished with exit code 0
 
恭喜你,安裝成功,現在可以去加速訓練啦!!!

 

 

另外補充一下,如果你想要在anaconda里面使用tensorflow-gpu需要這樣操作:

1. 打開anaconda prompt

2. 輸入activate tensorflow-gpu,下次安裝其他模塊需要先激活tensorflow-gpu環境,在這個環境內安裝

3. jupyter notebook 打開anaconda瀏覽器頁面

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM