1. 卸载cudnn
1.1查询cudnn
sudo dpkg -l | grep cudnn
1.2卸载操作
cudnn-local-repo-ubuntu2004-8.9.5.29要用purge命令删除
sudo dpkg -r libcudnn8-samples
sudo dpkg -r libcudnn8-dev
sudo dpkg -r libcudnn8
sudo apt-get purge cudnn-local-repo-ubuntu2004-8.9.5.29
再次查询,没有输出则卸载完成。
2. 安装cudnn
2.1检查是否安装CUDA
先检查/usr/local目录下有没有cuda或cuda-xx.xx文件夹,如果没有,那99.99%说明该机器没有安装CUDA。
如果说CUDA是GPU深度学习的运行库,那么cuDNN就是训练加速工具,两者要相互配合使用,所以一般机器学习需要训练引擎(tensorflow-gpu)或Pytorch + CUDA + cuDNN使用。想不安装cuDNN是不可以的,而且cuDNN版本要和CUDA版本相互搭配。(参考)
nvidia-smi
nvcc --version
2.2安装参考
Installation Guide - NVIDIA Docs
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2