tensorflow2.1對應 GPU版本的安裝與卸載

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的安裝過程

選擇自定義安裝

不要選擇vs 集成

cuda 安裝成功了

三、cuDNN的安裝：

進入官網下載cudnn 需要註冊帳號，之後再下載cudnn

https://developer.nvidia.com/cudnn

下載完成之後應當是一個壓縮文件，解壓。

文件內容

然後將解壓縮後的文件複製到cuda安裝路徑下的cudnn文件夾(這個文件夾需要自己建)。如下圖：

四、環境變量的配置

配置Path的環境變量 cudnn的bin lib in

配置Path的環境變量 cudnn的bin lib include

命令行輸入

tf.test.is_gpu_available()

最後結果為True

六、卸載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

同樣把它刪掉，然後重啟，就刪除乾淨了