浏览量: 100 次浏览

Ubuntu16.04安装多版本cuda和cudnn

2020年2月25日 0 作者 Nie Hen

1 cuda8.0+cudnn6.0.5的安装见ubuntu16.04配置caffe环境。

2 安装cuda-9.0

cuda9.0安装地址

下载,按照Installation Instructions安装。

验证是否安装成功:nvcc -V,
出现cuda版本信息则表示安装成功。

3 安装cudnn-7.0.5

安装地址:https://developer.nvidia.com/rdp/cudnn-archive
(1)sudo chmod -R 777 cudnn-9.0-linux-x64-v7.tgz
(2)sudo tar zxvf cudnn-9.0-linux-x64-v7.tgz
(3)sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
(4)sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
(5)sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*

4 cuda版本的自由切换

命令行sudo gedit ~/.bashrc打开.bashrc,
也许你已经装过cuda8.0,则.bashrc下应该有如下行:

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

将所有的cuda-8.0更改为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"  

到这一步我们已经设置好了链接库,最后我们只需要在终端敲入:

sudo rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #创建新 cuda 的软链接

查看cuda版本:

nvcc -V,
可以发现由cuda9.0切换到cuda8.0。