Ubuntu Cuda版本切换

一、前言

windows cuda版本切换见：https://blog.csdn.net/luozhichengaichenlei/article/details/104678203；

因为我一直用cuda10.0，但是现在很多东西需要更新版本的cuda，比如tensorflow2.3就需要cuda10.1；

为了能够在工作中更方便，准备在电脑上安装两个及以上的cuda版本，然后应对不同的项目进行不同cuda版本切换；

 

二、步骤

参考网址：https://blog.csdn.net/qq_29300341/article/details/84573517

1、写博客之前，我的ubuntu18.04已经是装好了cuda10.0的，现在先下载cuda10.1的安装软件cuda_10.1.168_418.67_linux.run

 

2、安装命令

sudo bash ./cuda_10.1.168_418.67_linux.run

安装时不用装Driver和Sample，安装完成后你的/usr/local/下有这三个文件夹，cuda、cuda-10.0和cuda-10.1，其中cuda文件夹是软链接，链接到我新安装的cuda10.1；

 

3、环境配置

这步很重要，因为版本切换的原理就是软链接+环境export，用以下命令进行环境配置：

注意路径中一定是cuda，而不是cuda-10.0或者cuda10.1；

sudo gedit ~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH

source ~/.bashrc

 

4、版本切换

切换到10.1：

sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.1  /usr/local/cuda
nvcc -V

 

切换到10.0：

sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0  /usr/local/cuda
nvcc -V

 

三、其他

cudnn安装：

（1）下载cudnn-10.1-linux-x64-v7.6.5.32.tgz并解压，复制到路径，注意路径中一定是cuda-10.1，而不是cuda，这里和上面正好相反；

tar -zvxf cudnn-10.1-linux-x64-v7.6.5.32.tgz
sudo chmod 777 -R *

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64

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