本文將介紹 CUDA 和 cuDNN 的安裝與配置，包括 CUDA 版本的確定、CUDA 的安裝與配置、cuDNN 的安裝與配置。運(yùn)行 CUDA 應(yīng)用需要支持 CUDA 的 GPU、CUDA Toolkit 以及 CUDA Driver。

Running a CUDA application requires the system with at least one CUDA capable GPU and a driver that is compatible with the CUDA Toolkit.

前期準(zhǔn)備

確定 CUDA Toolkit 安裝版本

安裝 CUDA 實(shí)際是指安裝 CUDA Toolkit，安裝 CUDA Toolkit 前需要確定的是，哪些版本的 CUDA Toolkit 可以安裝。

參照 CUDA Documentation，找到 Table 3，（部分）如圖 1 所示，表格中列出了CUDA Toolkit 所需的最小 Driver Version，即所需最小 NVIDIA 驅(qū)動(dòng)程序版本。

圖1 CUDA Toolkit and Corresponding Driver Versions

查看 Driver Version 有兩種方式。

方式一：在 ?命令提示符? 查看。

在 ?命令提示符? 執(zhí)行 nvidia-smi，如圖 2 所示。NVIDIA-SMI 表示 nvidia-smi 工具的版本，Driver Version 表示 NVIDIA 驅(qū)動(dòng)程序版本，CUDA Version 表示當(dāng)前 Driver Version 所支持的最高 CUDA 版本。CUDA Version 的值可由 Driver Version 根據(jù) 上述 Table 3 推出。

圖2 命令提示符查看 Driver Version

方式二：在 NVIDIA 控制面板查看。

打開(kāi) NVIDIA 控制面板（NVIDIA Control Panel），點(diǎn)擊 ?系統(tǒng)信息? 或者 ?幫助? 下的 ?系統(tǒng)信息?，如圖 3 所示。

圖3 NVIDIA 控制面板

在 ?顯示? 中可查看 CUDA Version，在 ?組件? 中可查看 Driver Version，如圖 4 和圖 5 所示。

圖4 NVIDIA 控制面板——顯示

圖5 NVIDIA 控制面板——組件

題外話，有些帖子以 CUDA Version 為依據(jù)，并且稱：CUDA Version 是 x.y 的話，就安裝版本為 x.y.z 的 CUDA。這句話不完全對(duì)，也可以安裝版本小于 x.y 的 CUDA，CUDA Version 當(dāng)前所支持的最高 CUDA 版本。

注冊(cè) NVIDIA 賬號(hào)

由于下載 cuDNN 需要登錄，所以需要先訪問(wèn) NVIDIA 官網(wǎng)，使用郵箱注冊(cè)賬號(hào)。

CUDA 的安裝與配置

下載

訪問(wèn) cuda-toolkit-archive，如圖 6 所示，下載準(zhǔn)備階段確定的版本。

圖6 選擇 CUDA Toolkit 版本

點(diǎn)擊選中的版本，根據(jù)操作系統(tǒng)進(jìn)行選擇，點(diǎn)擊 Download?！癳xe(local)” 包括安裝過(guò)程所需所有資源，“exe(network)” 在安裝過(guò)程中需要聯(lián)網(wǎng)下載某些資源，建議 “exe(local)” ，如圖 7 所示。

圖7 CUDA Toolkit 下載

安裝

我下載的是 “exe(local)” ，第一步中的路徑是安裝過(guò)程中臨時(shí)文件的存放路徑，保持默認(rèn)就可以。

圖8 CUDA Toolkit 安裝-選擇臨時(shí)文件存放路徑

若提示“您正在安裝老版本的驅(qū)動(dòng)程序。系統(tǒng)可在計(jì)算機(jī)定位或未定位時(shí)安裝新版本的驅(qū)動(dòng)程序”，說(shuō)明下載的安裝包中包含的驅(qū)動(dòng)比現(xiàn)有驅(qū)動(dòng)舊，可忽略，直接點(diǎn)擊 ?繼續(xù)?。

NVIDIA 驅(qū)動(dòng)下載地址 NVIDIA GeForce 驅(qū)動(dòng)程序 - N 卡驅(qū)動(dòng) | NVIDIA。

圖9 CUDA Toolkit 安裝-版本提示

同意 NVIDIA 軟件許可協(xié)議。

圖10 CUDA Toolkit 安裝-軟件許可協(xié)議

?精簡(jiǎn)? 從說(shuō)明中可以看出，這其實(shí)不是“精簡(jiǎn)”，而是“完整”，會(huì)安裝所有組件并將現(xiàn)有顯卡驅(qū)動(dòng)覆蓋，翻譯成“全家桶”可能更合適。對(duì)本文而言，許多組件并不需要，故推薦選擇 ?自定義?，以下會(huì)分別演示這兩種方式。

圖11 CUDA Toolkit 安裝-安裝選項(xiàng)

精簡(jiǎn)安裝

選擇 ?精簡(jiǎn)?，點(diǎn)擊 ?下一步?。

若出現(xiàn)如下提示，即如圖 12 所示，可選擇繼續(xù)安裝，如果安裝失敗，則改用 ?自定義? 安裝方式。

NO supported version of Visual Studio was found. Some components of the CUDA Toolkit will not work properly. Please install Visual Studio first to get the full functionality.

圖12 CUDA Toolkit 安裝-提示缺少 Visual Studio

點(diǎn)擊 ?下一步?。

圖13 CUDA Toolkit 安裝-確定安裝項(xiàng)

取消勾選。

圖14 CUDA Toolkit 安裝-安裝完成

自定義安裝

選擇 ?自定義?，點(diǎn)擊 ?下一步?。僅勾選 CUDA 下的 Development 和 Runtime 即可。

圖15 CUDA Toolkit 安裝-自定義安裝選項(xiàng)

指定安裝路徑。

圖16 CUDA Toolkit 安裝-選擇安裝位置

安裝成功。

圖17 CUDA Toolkit 安裝-安裝完成

檢查

在 ?命令提示符? 執(zhí)行 nvcc -V ，若顯示版本，則說(shuō)明安裝成功，若無(wú)顯示，先檢查是否配置環(huán)境變量。

配置

在 ?系統(tǒng)變量? 下的 Path 變量中檢查是否存在 CUDA 安裝目錄下的 bin 和 libnvvp，不存在則需進(jìn)行添加，如圖 18 所示，圖中 CUDA 為前文在安裝過(guò)程中指定的 CUDA 的安裝目錄。

圖18 CUDA Toolkit 安裝-配置環(huán)境變量

若在 ?系統(tǒng)變量? 下檢查存在環(huán)境變量 CUDA_PATH ，其值為 CUDA 安裝路徑，如圖 19 所示，則可將圖中配置簡(jiǎn)化為。

%CUDA_PATH%\lib
%CUDA_PATH%\libnvpp

圖19 CUDA Toolkit 安裝-配置環(huán)境變量

cuDNN 的安裝與配置

訪問(wèn) cuDNN Archive ，根據(jù)操作系統(tǒng)和 CUDA 的版本進(jìn)行選擇和下載，如圖 20 所示

圖20 選擇 cuDNN 版本

下載的是一個(gè)壓縮包，解壓后如圖 21 所示。

圖21 cuDNN 解壓目錄

找到 CUDA 的安裝目錄，其下也存在 bin、include、lib 三個(gè)目錄，如圖所示。將 cuDNN 解壓后得到的 bin、include、lib 三個(gè)目錄中的內(nèi)容分別復(fù)制到 CUDA 安裝目錄下的對(duì)應(yīng)目錄中。

圖22 CUDA 相應(yīng)目錄

在 CUDA 的安裝目錄下的 extras\demo_suite 目錄下找到 bandwidthTest.exe 和 deviceQuery.exe 兩個(gè)文件，分別在 ?命令提示符? 中執(zhí)行（直接雙擊可能會(huì)一閃而過(guò)），如圖 23 和圖 24 所示，若均執(zhí)行成功，說(shuō)明 cuDNN 安裝成功。

圖23 執(zhí)行 bandwidthTest.exe

圖24 執(zhí)行 deviceQuery.exe

END

以上就是本文的全部?jī)?nèi)容，文檔會(huì)根據(jù)自己的實(shí)際使用和各位提出的問(wèn)題而不斷更新。

如果覺(jué)得本文對(duì)您有一點(diǎn)點(diǎn)幫助，歡迎點(diǎn)贊、轉(zhuǎn)發(fā)加關(guān)注，這會(huì)對(duì)我有非常大的幫助，如果有任何問(wèn)題，歡迎在評(píng)論區(qū)留言，咱們下期見(jiàn)！

文章文檔：公眾號(hào) 字節(jié)幺零二四 回復(fù)關(guān)鍵字即可獲取本文最新文檔，關(guān)鍵字放在公眾號(hào)的本文中。