Ubuntu下安裝CUDA

鏡像下載、功能變數名稱解析、時間同步請點擊 阿裡雲開源鏡像站

註意：
Ubuntu18.04預設GCC-7，由於CUDA 9.x不支持GCC-7（下載頁面沒有對ubuntu18的支持），所以需要安裝低版本的5或者<= 6.3.0，並設置為預設版本。否則應安裝10.0以上版本CUDA

1.查看是否安裝了cuda

# 法1
cat  /usr/local/cuda/version.txt
# 法2
nvcc --version

2.若沒有安裝，則查看是否有N卡驅動，若無N卡驅動，則到軟體與更新 -> 附加驅動中安裝驅動

3.查看N卡驅動支持的cuda版本

nvidia-smi

如下圖支持最高CUDA版本是11.4

4.到cuda-toolkit-archive，選擇需要的CUDA下載

5.如下圖，選擇runfile(local)，並使用生成的指令進行下載和安裝

若第1步提示Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.，選擇continue，在下一步中去除driver項，之後選擇install：

安裝完成後，顯示如下：

6.在~/.bashrc文件中添加如下環境變數：

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

7.驗證是否安裝成功

nvcc -V

8.安裝cuDNN，到cudnn-archive下載和CUDA對應的版本

1.Navigate to your <cudnnpath> directory containing the cuDNN tar file.
2.Unzip the cuDNN package.
  $ tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz
3. Copy the following files into the CUDA toolkit directory.
  $ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
  $ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
  $ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

按照↑教程，可下載cuDNN Library for Linux (x86_64)用複製的方式安裝，使用如下命令查看安裝版本

cat /usr/local/cuda-11.3/include/cudnn.h | grep CUDNN_MAJOR -A 2

9.也可下載Runtime Library和Developer Library的deb包，使用sudo dpkg -i xxx.deb命令進行安裝，使用如下命令查看安裝版本

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

10.多版本CUDA安裝重覆上述步驟即可，最後通過更改第6步中環境變數切換CUDA版本

本文轉自：https://blog.csdn.net/CC977/article/details/122789394