天天看點

window10系統下python3.5.0安裝tensorflow-gpu

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個版本一定要全部對應,防止出錯浪費時間

window10系統下python3.5.0安裝tensorflow-gpu

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後預設直接安裝,系統變量會出現幾個路徑

window10系統下python3.5.0安裝tensorflow-gpu

系統變量path路徑也會多出2個路徑:

window10系統下python3.5.0安裝tensorflow-gpu

cmd下輸入nvcc -V檢視是否安裝成功

window10系統下python3.5.0安裝tensorflow-gpu

下載下傳cudnn7版本檔案

檔案複制到對應的檔案名的檔案中

然後安裝pip install tensorflow-gpu==1.10.0

window10系統下python3.5.0安裝tensorflow-gpu

沒有錯誤提示安裝成功

檢查是否可以正常調用gpu跑程式:

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

window10系統下python3.5.0安裝tensorflow-gpu

出現這兩個名字就表示tensorflow-gpu版本的已經安裝成功了