参考链接:Ubuntu 多版本Cuda(8.0,9.0)以及CuDnn安装 - 简书
1.先下载自己想要的cuda版本
2.下载对应cudnn版本cudnn
网上都有很多安装教程,本文主要将建立软连接
3.配置cuda环境
在 ~/.bashrc末尾添加路径(在终端输入:vim ~/.bashrc )
#在文件结尾处添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
添加后,按ESC,再按shift + :,输入“wq!” 保存并退出
4.建立软连接
多个cuda版本切换
安装完cuda9.0过后在 /usr/local/下看的如下:
这里,cuda-11.7和cuda-11.8就是我们安装的两个cuda版本了,而cuda是一个软链接,它指向我们指定的cuda版本(之前是cuda-11.7,现在想要指向11.8)
可以使用stat命令查看当前cuda软链接指向的哪个cuda版本,在/usr/local/目录打开终端,如下所示:
可以看到现在笔者的cuda是指向的cuda11.7,再用nvcc --version 命令看下我们当前使用的是哪个版本.
当我们想使用cuda-11.8版本时,只需要删除上面的软链接,然后重新建立指向cuda-8.0版本的软链接即可(注意名称还是cuda,因为要与bashrc文件里设置的保持一致)
再在终端输入: nvcc --version 就可以看到切换成功了。
pytorch 链接:Previous PyTorch Versions | PyTorch