Win10環境下配置CUDA與cuDNN

2018-09-20 置頂文章

61312
0
CUDA
2020-06-15

NVIDIA的顯示卡可以加速訓練速度，

這次為了跑Tensorflow

需要配置CUDA，來使用顯示卡加速。

配置內容的相關資訊：
NVIDIA GTX 760
原版驅動程式：375.63 → 更新：411.63
作業系統：WINDOWS 10 64 bit
CUDA：Toolkit 9.0
cuDNN：cuDNN v7.2.1 (August 7, 2018), for CUDA 9.0
VS：Visual Studio Community 2017 (version 15.8)

安裝項目

1. NVIDIA
先檢查電腦內之驅動版本是否高於「376.51」，低於請下載最新的版本來更新驅動！
驅動程式下載：https://www.nvidia.com.tw/Download/index.aspx?lang=tw

控制台/硬體和音效/NVDIA控制面板點進去就可以看到驅動版本了！

2. CUDA
安裝時注意，請選擇自定安裝，因為已經自行安裝較高版本的驅動，所以只需要勾選CUDA即可。
CUDA下載：https://developer.nvidia.com/cuda-toolkit-archive

(安裝位置請別更改，以免有錯)
(如果之前已有安裝vistual studio請在背景把它關掉不然會跟安裝CUDA衝突)

3. cuDNN
請根據你的CUDA版本選擇對應版本下載！
解壓縮後，將三個資料夾中的檔案，個別複製到CUDA的相對應資料夾中。
cuDNN下載：https://developer.nvidia.com/cuda-toolkit-archive

在這裡因為我安裝的是CUDA Toolkit 9.0 所以選對應的進行下載。

下載解壓縮後，可以看到三個資料夾

請把cuDNN三個資料夾中的檔案，個別複製加進CUDA對應的資料夾中即可。

例如：我CUDA的路徑為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

所以就分別把cuDNN資料夾中的bin、include、lib中的檔案

複製加進CUDA路徑中各自對應的資料夾裡即可！

3.VS
visual studio下載：https://my.visualstudio.com/Downloads/Featured

選擇想要的安裝路徑，選擇自定安裝，只勾選安裝必要的「Visual C++」，其他功能若有需要請自行安裝

因為我電腦本身就沒有載過VS，所以選擇2017最新版的進行安裝
電腦已經有舊版本的可以下載Update版本來進行更新。

環境變數設定

進入控制台/系統及安全性/進階系統設定/進階/環境變數
尋找「系統變數System」中的「Path」，新增以下變數：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Tools\MSVC\14.15.26726\bin
(VS 根據版本路徑會有不一樣，請找到VC下的bin資料夾為原則)