Ubuntu16.04安装多个cuda、cudnn版本并自由切换

之前安装了cuda9.0+cudnn7.0,现在需要用cuda8.0+cudnn5.1

一、安装CUDA

1、下载cuda8.0

https://developer.nvidia.com/cuda-80-ga2-download-archive

2、打开终端,cd到.run文件所在目录

如果出现没有那个文件或目录,则去找到该.run文件,右键->属性,查看其路径。

3、终端输入:

sudo chmod a+x xxxx.run
sudo sh xxxx.run

出现下图时,按Q键跳过即可。

出现下图时,输入accept,回车。

出现下图时,输入n

出现下图时,输入y

出现下图,输入y

出现下图,按回车

出现下图,输入n

出现下图,输入y

出现下图,回车

安装中。。。。。

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==uploading.4e448015.gif转存失败重新上传取消 wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

二、安装CUDNN

1、下载cudnn5.1(需要注册账号)

https://developer.nvidia.com/rdp/cudnn-archive

2、cd进入到.tgz文件所在目录

3、解压.tgz文件

$ tar -zxvf xxxxxx.tgz

解压后,在与.tgz同文件夹内会生成一个cuda文件夹,里面有include和lib64两个文件夹

4、将include和lib64两个文件夹复制到/usr/local/cuda-8.0(cuda-8.0的安装路径)中

sudo cp {include,lib64} /usr/local/cuda-8.0

如果出现“略过目录”,则使用:

sudo cp -r {include,lib64} /usr/local/cuda-8.0

三、环境变量修改

之前安装CUDA8.0的时候已经添加过环境变量,这里我们需要对系统变量稍作修改,以后切换CUDA版本的时候只需要修改CUDA目录软连接到想使用的CUDA版本目录就可以了(或者直接修改环境变量,变更CUDA版本的目录也可以)。接下来分别说明环境变量修改内容和切换步骤:

(1)终端输入:

gedit ~/.bashrc

(2)在打开的文件中找到以下几行:

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

修改为:

# added by cuda8.0
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

四、CUDA版本切换使用说明

(1)删除原来的软连接:

sudo rm -rf /usr/local/cuda

(2)建立新的CUDA软连接

sudo ln -s /usr/local/cuda-xxx /usr/local/cuda  #使用哪个版本xxx就是哪个版本号
发布了75 篇原创文章 · 获赞 9 · 访问量 9579

猜你喜欢

转载自blog.csdn.net/ly_twt/article/details/104905417