cuDNN安装教程
·
安装步骤
以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
更多推荐
所有评论(0)