2017年4月23日 星期日

於Win10環境下配置CUDA與cuDNN

nVIDIA的顯示卡常用在許多科學運算與機器學習的加速上。但是在此之前,需要配置CUDA,方可使用顯示卡加速。
網路上不乏各種教學文章,但多太過老舊或太過簡略,只好自己再來寫一篇。

基於GTX1060,以下為此次示範的軟體相關資訊:
作業系統:Windows 10 64bit
CUDA:8.0.61
cuDNN:cuDNN v5.1 Library for Windows 10 (Jan 20, 2017), for CUDA 8.0
VS:Visual Studio Community 2015 with Update 3 (English)

  • 下載檔案

 CUDA
 進入網頁後,如圖選擇,請注意,Installer Type這裡務必選擇local,根據經驗,使用network安裝容易少裝東西。


Visual Studio Community 2015 with Update 3 (English)
下載Visual Studio Community 2015 with Update 3 (English)需先加入開發者社群方可下載。
在這裡是選用x64,語言為英文的ISO檔,據說選用中文版會有一些小問題,不過我沒有用過所以也不知道就是了。
cuDNN
下載cuDNN前需先加入nvidia developer社群,登入後方可下載。
由於CUDA是使用8.0版,故cuDNN請選擇for CUDA 8.0。又因為有軟體尚未支援cuDNN v6.0,本文以cuDNN v5.1做示範。請記得根據系統選擇您要的版本。


  • 安裝檔案

CUDA

安裝CUDA前請先檢查電腦內之驅動版本是否高於「376.51」,若否或是尚未安裝,請自行至nVIDIA官方網頁下載新版驅動並安裝,過低版本的驅動可能會有不可預期之錯誤。而我自己安裝CUDA自帶376.51版的驅動又出現許多問題,故建議自行安裝較高版本的驅動比較保險。連結如下:
nVIDIA驅動程式下載

點開CUDA解壓縮後進行安裝,首先會進入授權同意書的頁面,點選「同意並繼續」。

安裝選項選擇自訂

因為已經自行安裝較高版本的驅動,故這裡只需要溝選CUDA即可。

選取安裝位置,請盡量不要更動,下一步即可。

cuDNN
cuDNN下載為一壓縮檔,點擊檔案並進入CUDA資料夾後可以看到如下圖的3資料夾
將3個資料夾解壓縮至「C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0」中

Visual Studio Community 2015 with Update 3 (English)

點擊ISO掛載檔案後進入資料夾,點擊「vs_community.exe」進行安裝。


Choose your installation location的部份選擇想要的安裝路徑,在此以「C:\Program Files (x86)\Microsoft Visual Studio 14.0」作為範例,並在Choose the type of installation的部份選擇Custom,點擊Next。

在這裡我只勾選安裝必要的「Visual C++」,其他功能若有需要請自行安裝,點選Next。

表中會告訴你會安裝哪些東西,點選Install即可安裝。

  • 環境變數設定

進入控制台→系統及安全性→進階系統設定→進階→環境變數
尋找「系統變數」中「Path」的部份並用左鍵雙擊,新增以下變數:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin

  • 檢查

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

恭喜你可以開始調用GPU來加速啦~





創用 CC 授權條款
本著作由Wei製作,以創用CC 姓名標示-非商業性 4.0 國際 授權條款釋出。

9 則留言:

  1. 感謝教學 這篇比以前的清楚很多 謝謝

    回覆刪除
  2. Visual Studio Community 2015 with Update 3 (English) 好像被移除了

    回覆刪除
    回覆
    1. 我剛剛查看還在啊?
      請問您有加入開發者社群當中嘛?
      雖然Visual Studio Community 2015 with Update 3 (English) 可以免費下載,但需要先加入開發者社群當中喔。

      刪除
  3. 原來真的是登入身分不完全 再登入一層 Community身分就看到了Visual Studio Community 2015 with Update 3 感謝
    不過還有一個問題 讓我裝這個了好幾次才試成功 (不過我不是很確定)
    我發現在我的電腦 順序好像要先裝VS 再裝cuda 及cudnn 才有效耶

    之前先裝cuda 及cudnn 再裝vs 的結果 nvcc -V好像都沒過

    回覆刪除
    回覆
    1. 說實話,我沒注意過這問題欸
      裝了好幾次好像也沒遇到這問題
      抱歉幫不了你QQ

      刪除
  4. 不好意思想請問一下,CUDA在17年6月之後有更新一個Patch2的安裝包,不下載那個會有問題嗎??

    回覆刪除
    回覆
    1. 根據官方網站的說法,Patch2主要用於修復Bug和提昇性能,裝一下應該會比較好。

      刪除