一,本機配置 Win10 64bit NVIDIA GeForce GTX 960M Python3.7(Anaconda) 二,安裝CUDA 親測,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我們可直接選擇cuda10.0的版本 Window10下載CUDA10 安裝步 ...
一,本機配置
Win10 64bit
NVIDIA GeForce GTX 960M
Python3.7(Anaconda)
二,安裝CUDA
親測,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我們可直接選擇cuda10.0的版本
安裝步驟為:
1)運行.exe可執行和層序。安裝軟體會先運行一個系統檢查,如果沒有軟硬體不相容的情況就能繼續進行下一步。若有不相容情況,系統檢查則會報錯,無法進行下一步。如果報錯了,就最先考慮下是不是自己的顯卡不被該版本支持。在安裝過程中,它會先自動安裝和你電腦對應的NVIDIA驅動。
2)選擇預設安裝就行。
三,安裝cuDNN
下載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())
出現下列結果表明安裝成功