安装步骤
以CUDA 12.8为例,安装对应的 cuDNN,版本是 cuDNN 8.9.x。

一、下载 cuDNN

  • 访问 NVIDIA cuDNN 官方网站
  • 注册并登录 NVIDIA 开发者账号。
  • 选择 cuDNN v8.9.1 (May 5th, 2023), for CUDA 12.x。
  • 下载 Local Installer for Ubuntu22.04 x86_64 (Deb) 或 Tar File(通用手动安装包)。

二、安装 cuDNN

如果下载的是 .deb 文件,使用以下命令安装:

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.1.23_1.0-1_amd64.deb
sudo apt-get update
sudo apt-get install libcudnn8=8.9.1.23-1+cuda12.1

如果下载的是 .tar.xz 文件,手动安装:

tar -xvf cudnn-linux-x86_64-8.9.1.23_cuda12-archive.tar.xz
cd cudnn-linux-x86_64-8.9.1.23_cuda12-archive
sudo cp include/cudnn*.h /usr/local/cuda/include
sudo cp lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

三、验证安装

安装完成后,运行以下命令验证 cuDNN 是否安装成功:

cat /usr/local/cuda/lib64/libcudnn.so
Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐