显卡驱动安装

• nvidia 官网找到显卡适配的驱动： https://www.nvidia.com/Download/index.aspx
  • Installer Type 选择 runfile (local)
• 禁用 nouveau

  sudo gedit /etc/modprobe.d/blacklist.conf # 在blacklist.conf文件末尾加上 blacklist nouveau
  sudo update-initramfs -u # 更新
  reboot 
  lsmod | grep nouveau # 没有输出说明禁用成功
  

• 安装依赖

  sudo apt-get update
  sudo apt-get install gcc
  sudo apt-get install ubuntu-make
  sudo apt-get install make
  

• 安装驱动

  sudo apt-get remove –purge nvidia* # 删除旧版本驱动
  sudo chmod a+x NVIDIA-Linux-x86_64-515.57.run # 添加可执行权限
  sudo ./NVIDIA-Linux-x86_64-515.57.run -no-x-check -no-nouveau-check -no-opengl-files # 以下两个选择与默认相反，其他选择与默认一致
  # 1. Install NVIDIA’s 32-bit compatibility libraries 选择与默认相反的选项： no
  # 2. Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择与默认相反选项： yes
  

• 检验是否安装成功

  nvidia-smi # 有显存占用率等信息的表格出现代表安装成功
  

CUDA 11.7 安装

• nvidia 官网找到 CUDA 安装包： https://developer.nvidia.com/cuda-downloads

  • Installer Type 选择 runfile (local)
    

• 安装依赖

  sudo apt-get install build-essential
  

• 下载安装包并安装

  wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
  sudo sh cuda_11.7.0_515.43.04_linux.run
  

cuDNN 8.4.1.50 安装

• 前往 cuDNN 官网下载源文件： https://developer.nvidia.com/rdp/cudnn-download
  
• 解压并复制 cuDNN 库文件到 CUDA 的库文件目录下（注意：这里也可以单独给 cuDNN 设置到一个单独目录下并添加环境变量，更利于多版本管理）

  mkdir cudnn_8.4.1.50
  tar xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz -C cudnn_8.4.1.50
  cd cudnn_8.4.1.50
  sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 注意看下 cudnn.h 目录结构是否正确
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
  sudo chmod a+r /usr/local/cuda/include/cudnn.h 
  sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
  

多 CUDA 版本共存 （CUDA11.3 安装）

• 官网下载安装包：
  • https://developer.nvidia.com/cuda-11.3.0-download-archive?target_os=Linux
    
  • 参考以上 CUDA 11.7 的安装方式进行安装
  • 修改 /usr/local/cuda 到需要的版本即可