【经验】linux下cuda的更换

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-archive
	sudo 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