修改与之前修改nvcc、cuda这些的原理是一样的。
国内镜像
https://mirrors.aliyun.com/nvidia-cuda
检查系统版本
source /etc/os-release && echo $VERSION_ID
导入apt仓库和其GPG key
- 阿里云镜像源
#第一步若输出 16.04
curl -fsSL https://mirrors.aliyun.com/nvidia-cuda/ubuntu1604/x86_64/3bf863cc.pub | apt-key add -
echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1604/x86_64/ /" > /etc/apt/sources.list.d/cuda.list
#第一步若输出 18.04
curl -fsSL https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/3bf863cc.pub | apt-key add -
echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/ /" > /etc/apt/sources.list.d/cuda.list
#第一步若输出 20.04
curl -fsSL https://mirrors.aliyun.com/nvidia-cuda/ubuntu2004/x86_64/3bf863cc.pub | apt-key add -
echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu2004/x86_64/ /" > /etc/apt/sources.list.d/cuda.list
更新 apt 源:
apt update
卸载原来的cudnn
先查 cudnn 版本,
dpkg -l | grep cudnn
输入下面指令卸载 cudnn
# 卸载 cudnn7
dpkg --purge libcudnn7 libcudnn7-dev
# 卸载 cudnn8
dpkg --purge libcudnn8 libcudnn8-dev
用命令安装cudnn
根据自己需要安装对应版本的 cudnn。
比如你需要安装 libcudnn 8.5.0 版本,首先到 阿里云源上搜索、找到对应版本的具体版本号,这个8.5.0具体版本号就是8.5.0.96-1+cuda11.7
。
然后一行行输入下面指令安装:
apt-get install -y --no-install-recommends libcudnn8=8.5.0.96-1+cuda11.7 libcudnn8-dev=8.5.0.96-1+cuda11.7
apt-mark hold libcudnn8
再输入下面指令检查cudnn版本:
dpkg -l | grep cudnn