Ubuntu多cuda+cudnn环境

同一台机子可以兼容多个版本的CUDA的，这两天就要这么来回折腾，稍显麻烦。

1. 查看当前机子上CUDA版本；

nvcc --version

由此可以看到当前是CUDA 8.0。

2. 修改.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

把以上这种添加变量的方式改为如下：

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

那么下面切换为CUDA 9.0版本。

2. 从官网下载CUDA 9.0的run文件，安装过程中，选择不建立软连接。

3.  切换CUDA 9.0，使用下面命令。

sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
 

Cudnn照旧安装

安装cuDNN

把头文件和库文件拷贝到cuda的安装目录中，首先确保cuda的安装目录是/usr/local/cuda/，之前安装cuda时，已经默认安装目录为/usr/local/cuda/了

sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
————————————————
版权声明：本文为CSDN博主「songer93」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/songer93/article/details/95760018

 

网址

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

下载  

cuDNN v5.1 Library for Linux

cuda版本

cat /usr/local/cuda/version.txt

cudnn版本

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

 

 

 

这时就可以看到当前的CUDA版本变为9.0了。
---------------------
作者：mumoDM
来源：CSDN
原文：https://blog.csdn.net/mumoDM/article/details/79462604
版权声明：本文为博主原创文章，转载请附上博文链接！