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())
結果如下:
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"
]
另外補充一下,如果你想要在anaconda里面使用tensorflow-gpu需要這樣操作:
1. 打開anaconda prompt
2. 輸入activate tensorflow-gpu,下次安裝其他模塊需要先激活tensorflow-gpu環境,在這個環境內安裝
3. jupyter notebook 打開anaconda瀏覽器頁面