在 Ubuntu 20.04 上升级并验证 cuDNN 9.7.1 —— 基于 CUDA 12 环境指南

在 Ubuntu 20.04 上升级并验证 cuDNN 9.7.1 —— 基于 CUDA 12 环境指南

本文将系统地介绍如何在 Ubuntu 20.04 上通过 NVIDIA 官方本地仓库包，将系统范围内的 cuDNN 升级至 9.7.1，并在 CUDA 12.x 环境中完成安装验证。全程以 Conda/系统二者通用方式展开，不依赖于手动编译，适合新手与运维工程师参考。

---

目录

1. 前言
2. 下载 cuDNN 本地仓库包
3. 安装本地仓库并导入公钥
4. 更新 APT 索引
5. 安装针对 CUDA 12 的 cuDNN 9.7.1
6. 验证安装结果
7. 结语

---

前言

cuDNN（CUDA Deep Neural Network library）是深度学习框架在 NVIDIA GPU 上进行高性能计算的核心依赖。官方推荐为每个 CUDA 次版本使用与之匹配的 cuDNN release。本文以 CUDA 12.x（包括 12.8）为例，介绍如何安装 cuDNN 9.7.1。

---

下载 cuDNN 本地仓库包

1. 登录 NVIDIA Developer 官网 ，找到 cuDNN Library for Linux 对应 v9.7.1 的本地安装包。
2. 在终端执行：

   wget https://developer.download.nvidia.com/compute/cudnn/9.7.1/local_installers/cudnn-local-repo-ubuntu2004-9.7.1_1.0-1_amd64.deb
   

---

安装本地仓库并导入公钥

1. 安装 .deb 本地仓库包：

   sudo dpkg -i cudnn-local-repo-ubuntu2004-9.7.1_1.0-1_amd64.deb
   

2. 将仓库签名公钥复制至系统信任区：

   sudo cp /var/cudnn-local-repo-ubuntu2004-9.7.1/cudnn-*-keyring.gpg /usr/share/keyrings/
   

---

更新 APT 索引

sudo apt update

该命令会将刚刚安装的本地 cuDNN 仓库纳入 APT 源列表。

---

安装针对 CUDA 12 的 cuDNN 9.7.1

执行以下命令，以安装面向 CUDA 12.x 的运行时、开发头文件及示例包：

sudo apt-get -y install cudnn-cuda-12

系统会自动选择 cudnn9-cuda-12、libcudnn9-dev-cuda-12 及 libcudnn9-static-cuda-12 等对应包，并移除旧版 9.x（如 9.2.0）包。

---

验证安装结果

1. 查看头文件版本宏

   grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL" \
     /usr/local/cuda-12.8/targets/x86_64-linux/include/cudnn_version_v9.h
   

   若输出：

   #define CUDNN_MAJOR       9
   #define CUDNN_MINOR       7
   #define CUDNN_PATCHLEVEL  1
   #define CUDNN_VERSION     (CUDNN_MAJOR * 10000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
   

   则表明头文件已正确指向 9.7.1。

2. 在 PyTorch 中验证

   import torch
   
   print("CUDA available:", torch.cuda.is_available())
   print("cuDNN enabled:", torch.backends.cudnn.is_available())
   print("cuDNN version:", torch.backends.cudnn.version())
   

   运行结果应类似：

   CUDA available: True
   cuDNN enabled: True
   cuDNN version: 9701
   

   其中 9701 对应 cuDNN 9.7.1。

---