1. 程式人生 > 其它 >Ubuntu18.04安裝CUDA10.2+cuDNN

Ubuntu18.04安裝CUDA10.2+cuDNN

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