python版本:python3.5.0
顯示卡版本:gtx 780 (計算能力3.5)
系統:win10 64位
cuda版本:cuda9
cudnn版本:cudnn7版本
tensorflow-gpu版本:tensorflow-gpu1.10.0
vs版本:安裝visual studio 2015(旗艦版)
注意事項:
1:cuda,cudnn,tensorflow,python,這4個版本一定要全部對應,防止出錯浪費時間
2:tensorflow-gpu1.10.0需要gpu計算能力>=3.5,大于tensorflow1.11.0,如tensorflow1.12.0的就需要計算能力3.7以上(血的教訓,我的剛好3.5)。
3:安裝cuda後要配置系統環境變量,cudnn内容要複制到cuda對應的檔案夾中
4:最好檢查用驅動精靈,360驅動更新顯示卡驅動到最新版本
步驟
一:安裝python3.5.0版本,并配置環境變量。更新pip和setuptools兩個工具包
下載下傳cuda9.0後預設直接安裝,系統變量會出現幾個路徑
系統變量path路徑也會多出2個路徑:
cmd下輸入nvcc -V檢視是否安裝成功
下載下傳cudnn7版本檔案
檔案複制到對應的檔案名的檔案中
然後安裝pip install tensorflow-gpu==1.10.0
沒有錯誤提示安裝成功
檢查是否可以正常調用gpu跑程式:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
出現這兩個名字就表示tensorflow-gpu版本的已經安裝成功了