Ubuntu>卸载-安装cudnn

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

安装cuDNN-7.5.0-CSDN博客

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值