Win10環境下配置CUDA與cuDNN

  • 61315
  • 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資料夾為原則)

檢查

進入CMD並輸入「nvcc -V」檢查是否安裝成功。

有出現資訊表示成功!

 

如果失敗,請注意:

1.檢查環境變數路徑是否正確
2.重新開機再打開CMD測試
3.請重新確定CUDA及cuDNN的版本有相對應,沒有裝錯!

以上如果還是無法執行....

請重新安裝CUDA
(當初第一次失敗,重裝了之後就成功抓到...原因不明阿)

 

 


參考資料來源:https://rreadmorebooks.blogspot.com/2017/04/win10cudacudnn.html


 


人生美好~別浪費腦容量記程式碼 :- ) 

作者:CYL
出處:http://dotblogs.com.tw/cylcode
資料來源都會特別註明,有興趣都可查詢原出處,本站皆經過整理才分享,如有轉載請顯示出處及作者,感謝。