linux下cuda的更换
查看当前cuda和cudnn的版本
nvcc -V
cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
下载对应版本的cuda
查看驱动版本535.54.03
下载对应的cuda版本
版本查看https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
下载地址https://developer.nvidia.com/cuda-toolkit-archive
uname -a
我这里是centos系统,查看centos版本
cat /etc/centos-release
显示系统信息
按照版本选择下载,根据索引下载
更换cuda
rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
nvcc --version #查看当前 cuda 版本
下载安装cudnn
https://developer.nvidia.com/cudnn-archive
下载Local Installer for Linux x86_64(Tar)
解压缩
tar -xvf cudnn-11.0-linux-x64-v8.0.2.39.tgz
cd cudnn-linux-x86_64-8.4.0.27_cuda11.6-archivesudo cp lib/* /usr/local/cuda-11.2/lib64/sudo cp include/* /usr/local/cuda-11.2/include/sudo chmod a+r /usr/local/cuda/lib64/*sudo chmod a+r /usr/local/cuda/include/*
验证安装是否成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2