在Ubuntu上部署CUDA 11.5 和 cuDNN（P102-100）

以下是基于搜索结果整理的在 Ubuntu 上在线部署 CUDA 11.5 及其对应的 cuDNN 的步骤：

1. 安装 NVIDIA 驱动

• 添加 NVIDIA 驱动的 PPA 源：

  sudo add-apt-repository ppa:graphics-drivers/ppa
  sudo apt update
  

• 安装适合 CUDA 11.5 的 NVIDIA 驱动，如 nvidia-driver-460：

  sudo apt install nvidia-driver-460
  

• 安装完成后，重启系统以加载驱动。

2. 安装 CUDA 11.5

• 下载 CUDA 11.5 的安装包：

  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
  sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
  

• 添加 CUDA 的 APT 仓库：

  wget https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda-repo-ubuntu2004-11-5-local_11.5.0-470.57.02-1_amd64.deb
  sudo dpkg -i cuda-repo-ubuntu2004-11-5-local_11-5.0-470.57.02-1_amd64.deb
  sudo apt-key add /var/cuda-repo-ubuntu2004-11-5-local/7fa2af80.pub
  

• 更新 APT 索引并安装 CUDA：

  sudo apt update
  sudo apt install cuda-11-5
  

• 配置 CUDA 的环境变量：

  echo 'export PATH=/usr/local/cuda-11.5/bin:$PATH' >> ~/.bashrc
  echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.5/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
  source ~/.bashrc
  

3. 安装 cuDNN

• 访问 cuDNN 官网 并登录 NVIDIA 账号。

• 下载与 CUDA 11.5 对应的 cuDNN 版本，如 cuDNN 8.3.2。

• 解压下载的文件：

  tar -xvf cudnn-linux-x86_64-8.3.2.44_cuda11-archive.tar.xz
  

• 将解压后的文件复制到 CUDA Toolkit 的目录下：

  sudo cp cudnn-linux-x86_64-8.3.2.44_cuda11-archive/include/cudnn*.h /usr/local/cuda-11.5/include
  sudo cp -P cudnn-linux-x86_64-8.3.2.44_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.5/lib64
  sudo chmod a+r /usr/local/cuda-11.5/include/cudnn*.h /usr/local/cuda-11.5/lib64/libcudnn*
  

4. 验证安装

• 验证 CUDA 版本：

  nvcc --version
  

• 验证 cuDNN 版本：

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

以上步骤基于搜索结果整理，确保了在 Ubuntu 上在线部署 CUDA 11.5 及其对应的 cuDNN 的完整流程