GTX965M顯卡,win10系統
一、檢查tensorflow的Version
假定tensorflow 安裝完成,安裝 tensorflow 請參考 集成環境,安裝超簡單
通過 命令行
python
import tensorflow as tf
tf.__version__
如下圖:
二、CUDA的安裝
安裝cuda的版本 是根據tensorflow的版本來確定的
https://developer.nvidia.com/cuda-downloads
一般 2.0 tf對應10.0的cuda,2.1 tf對應10.1的cuda版本,請選擇合適的版本,本例中選擇version 10.1
cuda的安裝過程
cuda 安裝成功了
三、cuDNN的安裝:
進入官網下載cudnn 需要註冊帳號,之後再下載cudnn
https://developer.nvidia.com/cudnn
下載完成之後應當是一個壓縮文件,解壓。
然後將解壓縮後的文件複製到cuda安裝路徑下的cudnn文件夾(這個文件夾需要自己建)。如下圖:
四、環境變量的配置
配置Path的環境變量 cudnn的bin lib include
命令行輸入
tf.test.is_gpu_available()
六、卸載GPU
若安裝版本不對,卸載時正常情況不會把驅動卸載乾淨,導致重新裝失敗
若想卸載乾淨,重啟電腦後進入安全模式,為什麼進入安全模式(安全模式不用顯卡)
在位置
C:\Program Files\NVIDIA Corporation
C:\Program Files\NVIDIA GPU Computing Toolkit
默認安裝就是在這個位置,刪除這兩個文件夾。
然後再win+R運行,輸入regedit打開註冊表,找到下面這兩項
HKEY_LOCAL_MACHINE\SOFTWARE\NVIDIA Corporation
HKEY_LOCAL_MACHINE\SOFTWARE\NVIDIA GPU Computing Toolkit
同樣把它刪掉,然後重啟,就刪除乾淨了