- 官网下载指定的 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.4.1
-
sudo ln -sf libcudnn.so.7.4.1 libcudnn.so.7
-
sudo ln -sf libcudnn.so.7 libcudnn.so
-
sudo ldconfig