Ubuntu 配置多个版本cuda(10.0、10.1)

Ubuntu 安装CUDA

Ubuntu 安装CUDA_ubuntu安装cuda_Cynthia-石头的博客-CSDN博客

很多刚接触深度学习的朋友面临的第一个大坎就是开发环境的配置，经常一折腾就是一两天，更有甚至十天半月也不一定能让代码跑起来，实在是大大滴不友好。

环境配置问题一般都是各个框架与cuda版本的适配问题，装不好cuda就没办法调用gpu进行快速训练或推理，所以深度学习要是用不上gpu这不相当于学习了个寂寞！多提一句，很多童鞋一开始就是接触的tensorflow，特别在tf1.x时代，各个tf版本与不同cuda版本的适配，经常被搞得欲仙欲死，直接从入门到放弃。

今天给大家简单介绍一下如何在Ubuntu系统安装多个版本的cuda，有需要时顺滑进行切换，非常方便。

cuda、cudnn官方下载链接：

CUDA Toolkit Archive | NVIDIA Developer

在官网下载不同版本的cuda版本以及对应的cudnn版本，cudnn需要先注册一下。

这里最好下载runfile的文件格式

下载好的文件如下，这里以cuda10.1为例：

运行以下命令开始安装

sudo chmod +x cuda_10.1.105_418.39_linux.run
sudo ./cuda_10.1.105_418.39_linux.run

1、选择accept

2、按回车选择不安装驱动，然后选择Install进行安装

3、选择默认安装路径

# 设置当前用户的环境变量
sudo gedit ~/.bashrc

# 在.bashrc文件后面添加如下内容，注意这里不指定具体的cuda版本，就是为了方便切换

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

# 终端运行，使之生效

source ~/.bashrc

检查安装是否成功，终端输入ncvv --version

这里的cuda已经指向刚安装的cuda10.1的版本，是因为刚安装的时候默认选择了。

安装cuda10.1对应的cudnn，如果目前的cuda软连接指向的是10.1，直接复制过去即可，步骤如下：

tar -zcvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

#查看cudnn版本

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

至此，cuda10.1已经安装完成了。假设该机子上已经安装了cuda10.0的版本，如何切换回去呢？

也非常简单。

# 删除目前的软链接
sudo rm -rf /usr/local/cuda

# 生成新的软链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda 

# 查看当前cuda的版本，或者用nvcc --version
cat /usr/local/cuda/version.txt

参考：

Ubuntu 18.10 下安装CUDA10/CUDA10.1_cuda 10.1 ubuntu 18_哈里哈气的博客-CSDN博客

CUDA10.0和CUDA10.1同时安装，Linux系统下如何切换_yimaoshu的博客-CSDN博客

Ubuntn16.04下cuda10.0和cuda9.0自由切换_安装了cuda10.0但是版本还是9_kelly1250230225的博客-CSDN博客

Ubuntu16.04下成功安装多版本的Cuda和切换Cuda环境 --Cuda9.0 & Cuda10.0_cuda9.0和cuda10.0_梦坠凡尘的博客-CSDN博客

转载自：深度觉醒 公众号

欢迎大家扫码关注！