Ubuntu18.04安裝CUDA10.2+cuDNN
阿新 • • 發佈:2021-10-18
Ubuntu18.04安裝CUDA10.2+cuDNN
1 安裝CUDA10.2
-
下載CUDA:CUDA-Toolkit-Archive,選擇CUDA Toolkit 10.2版本
-
選擇作業系統、架構、Distribution及其版本(這裡是Ubuntu18.04)和runfile(local)檔案格式
-
可以使用wget下載,亦可複製連結進行下載
-
下載完成後:
sudo sh cuda_10.2.89_440.33.01_linux.run
步驟說明:
(1)Do you accept the above EULA? accept
(2)CUDA Installer 取消Driver安裝(Enter選擇/取消)-> Install
- 配置環境變數
vim ~/.bashrc
新增:
點選檢視程式碼
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/lcoal/cuda-10.1/lib64:$LD_LIBRARY_PATH
- 生效:
source ~/.bashrc
- 驗證:
點選檢視程式碼
cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
出現如下輸出,則CUDA安裝成功
或nvcc -V
出現如下輸出,則CUDA安裝成功。
2 安裝cuDNN
-
下載cuDNN:cuDNN Download
-
選擇:I Agree To the Terms of the cuDNN Software License Agreement -> Archived cuDNN Releases
-
選擇適配Ubuntu、CUDA 10.2的最新版cuDNN v8.2.2 -> cuDNN Library for Linux (x86)
-
下載完成後,解壓,進入相應目錄,執行以下命令:
點選檢視程式碼
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include sudo cp cuda/include/cudnn_version.h /usr/local/cuda-10.1/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64 sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
- 驗證:cat /usr/local/cuda-10.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2