先登入账号
网址:https://developer.nvidia.com/cudnn
选择ubuntu20.04 x86_64(Deb)
在下载好文件的文件夹下打开终端
sudo apt-get install zlib1g
sudo dpkg -i cudnn-local-repo-${distro}-8.x.x.x_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
安装依赖
改称自己下载的cudnn 版本号
改称自己下载的cudnn 版本号和cuda版本号
安装完毕
2、cudnn验证:
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
报错如下:
解决方案:
sudo apt install libfreeimage3 libfreeimage-dev
解决完后,重新输入
make clean && make
./mnistCUDNN
结果:
cudnn安装检验完成