切换cuda版本

首先, 查看一下 CUDA/cuDNN安装教程

安装后任意版本后, 查看系统安装的Cuda版本

ls -l /usr/local

可以看到安装了cuda-11.8, 并且默认cuda指向了11.8的cuda, 删除默认的cuda链接

sudo rm -rf /usr/local/cuda

安装新的cuda, 如12.6.1. 打开 CUDA Toolkit Archive 选中需要安装的版本

复制安装代码

wget https://developer.download.nvidia.com/compute/cuda/12.6.1/local_installers/cuda_12.6.1_560.35.03_linux.run
sudo sh cuda_12.6.1_560.35.03_linux.run

接受安装协议

查看cuda链接情况, 可以看到安装后cuda已经自动指向了新的cuda版本

编辑一下.bashrc 文件

nano /root/.bashrc

在最下面, 把环境变量修改为新的cuda版本 并保存

export PATH=/usr/local/cuda-12.6/bin${PATH:+:${PATH}}

刷新一下环境变量

source ~/.bashrc

查看一下cuda版本

nvcc -V

可以看到cuda已经切换到了12.6了

如果要切换回来11.8, 首先删除cuda链接

sudo rm -rf /usr/local/cuda

重新链接11.8的cuda

sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda
ls -l /usr/local

编辑环境变量

nano /root/.bashrc

修改回11.8

export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}

刷新环境变量

source ~/.bashrc

检查cuda版本

nvcc -V

可以看到已经切换到11.8了

切换cuDNN版本

由于不同的大版本的cuDNN和cuda不通用, 所以cuDNN需要重新安装

重复这篇教程 的后半部分即可

想要切换回之前的cuDNN, 在切换cuda版本后执行

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

检查cuDNN版本

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

可以看到已经切换回之前的cuDNN版本了