cuda 安裝#
1. cuda 的安裝#
前往 https://developer.nvidia.com/cuda-toolkit 下載。在安裝時一定要自訂安裝,否則將會安裝許多無用的東西。安裝選項中,可以選擇不更新驅動程式。
或者下載離線檔案進行安裝
安裝時,選擇自訂安裝。
安裝完成後,與英偉達 cuda 相關的程式如下圖所示。
注意,千萬不要勾選 Nsight Visual Studio Edition 2019.2 等類似的無用的東西。
2. 測試環境是否安裝成功#
執行 cmd,輸入nvcc --version
即可查看版本號;
set cuda
,可以查看 cuda 設定的環境變數。
安裝 cudnn#
cuDNN 地址如下,不過要注意的是,我們需要註冊一個帳號,才可以進入到下載界面。大家可以放心註冊的。
cuDNN Download | NVIDIA Developer
下載結果是一個壓縮包
cuDNN 配置#
cuDNN 叫配置更為準確,我們先把下載的 cuDNN 解壓縮,會得到下面的檔案:
解壓縮後的檔案:
下載後發現其實 cudnn 不是一個 exe 檔案,而是一個壓縮包,解壓縮後,有三個資料夾,把三個資料夾複製到 cuda 的安裝目錄下。
CUDA 的安裝路徑在前面截圖中有,或者打開電腦的環境變數查看,預設的安裝路徑如下:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\你安裝的cuda版本號
複製時看到,CUDA 的安裝目錄中,有和 cuDNN 解壓縮後的同名資料夾,這裡注意,不需要擔心,直接複製即可。cuDNN 解壓縮後的同名資料夾中的配置檔案會添加到 CUDA 安裝目錄中的同名資料夾中。【此處還是建議還是分別把資料夾的內容複製到對應資料夾中去】
cuDNN 其實是 CUDA 的一個補丁,專為深度學習運算進行優化的。然後再添加環境變數
首先把 cudnn 的同名資料夾複製到安裝 cuda 的位置
配置環境變數
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp