- 官网下载指定的 cudnn版本,选择 cuDNN Library for Linux
- 删除系统原来的的cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
- 解压下载的文件
tar zxvf cudnn-9.0-linux-x64-v7.5.0.56.tgz
- 到解压的文件夹下进行安装
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
- 建立链接,/usr/local/cuda/lib64/文件夹下执行以下命令,记得替换为自己的版本号
cd /usr/local/cuda/lib64/ sudo chmod +r libcudnn.so.7.5.0 sudo ln -sf libcudnn.so.7.5.0 libcudnn.so.7 sudo ln -sf libcudnn.so.7 libcudnn.so sudo ldconfig