配置 cuDNN 库（CUDA Deep Neural Network Library）

cuDNN 安装比 CUDA 简单，只需下载对应版本压缩包，将文件拷贝到指定目录，并设置正确权限即可。

1. 下载 cuDNN

  • 官方下载地址：NVIDIA cuDNN
  • 注意：可能需要注册 NVIDIA 账号并登录才能下载
  • 下载与 CUDA 版本匹配的 cuDNN
  • 以cudnn-linux-x86_64-9.16.0.29_cuda12-archive.tar.xz为例
    1.png

找到对应系统与架构的文件后，右键复制下载链接，使用wget进行下载，如：

wget https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.16.0.29_cuda12-archive.tar.xz

2. 上传并解压 tar 包

将下载好的 tar.xz 文件上传到服务器，然后解压：

# 假设文件名为 cudnn-linux-x86_64-9.16.0.29_cuda12-archive.tar.xz
tar -xvf cudnn-linux-x86_64-9.16.0.29_cuda12-archive.tar.xz

解压后会生成一个目录，例如：

cudnn-linux-x86_64-9.16.0.29_cuda12-archive/

3. 拷贝 cuDNN 文件到 CUDA 目录

# 拷贝头文件
cp cudnn-linux-x86_64-9.16.0.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include

# 拷贝库文件（保留符号链接）
cp -P cudnn-linux-x86_64-9.16.0.29_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64

4. 设置权限

chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

这样所有用户都可以读取 cuDNN 头文件和库文件。

5. 验证 cuDNN 安装

查看 cuDNN 版本是否正确：

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

输出类似：

#define CUDNN_MAJOR 9
#define CUDNN_MINOR 16
#define CUDNN_PATCHLEVEL 0

表示 cuDNN 9.16.0 已正确安装。

6. 注意事项

  1. 确保 CUDA 已正确安装并设置环境变量（PATH 与 LD_LIBRARY_PATH）
  2. cuDNN 版本必须与对应 CUDA 版本匹配
  3. 若多版本 CUDA 共存，请在切换到目标 CUDA 后再安装对应 cuDNN