win10安裝tensorflow-gpu1.13.1+cuda10.0+cudnn7.3.1


一,本機配置

Win10 64bit

NVIDIA GeForce GTX 960M

Python3.7(Anaconda)

二,安裝CUDA

親測,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我們可直接選擇cuda10.0的版本

Window10下載CUDA10

安裝步驟為:

1)運行.exe可執行和層序。安裝軟件會先運行一個系統檢查,如果沒有軟硬件不兼容的情況就能繼續進行下一步。若有不兼容情況,系統檢查則會報錯,無法進行下一步。如果報錯了,就最先考慮下是不是自己的顯卡不被該版本支持。在安裝過程中,它會先自動安裝和你電腦對應的NVIDIA驅動。
2)選擇默認安裝就行。

三,安裝cuDNN

cuDNN7.3.1下載

下載cuDNN需要自己先在以上網址(cuDNN官網)注冊個賬號,注冊成功后登陸進去選擇cuDNN7.3.1下載即可。

下載完成后是一個.zip包,解壓后進入目錄,會發現三個子目錄/bin,/include,/lib/x64。進去后會發現有三個文件,將這三個文件分別拷貝到cuda的安裝目錄的對應子目錄下

默認cuda的安裝目錄是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

四,配置系統Path環境變量

檢查你的系統Path環境變量中是否有這四項,沒有請手動添加

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

五,安裝tensorflow-gpu

打開Anaconda Prompt,輸入

pip install tensorflow-gpu

會自動下載並安裝tensorflow-gpu

六,安裝測試

打開pycharm,創建一個test.py文件,輸入

import tensorflow as tf

tf.enable_eager_execution()
print(tf.__version__)
hello = tf.constant("hello tensorflow")
print(hello.numpy())

出現下列結果表明安裝成功


免責聲明!

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



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