Ubuntu16.04下cuda和cudnn的卸载和升级

目录

1.cuda的升级

2.cudnn的升级

---

我一开始安装的是cuda8.0+cudnn6.0+tensorflow+mxnet，本来是没必要升级的，可是后来又需要安装caffe2，caffe2的官网教程中可以看到，如果使用conda安装的话，就只有cuda8.0+cudnn7.0或者cuda9.0+cudnn7.0了，而我又不想用源码安装caffe2，所以就只有一种办法了，那就是将当前的cuda和cudnn升级到cuda9.0+cudnn7.0。

1.cuda的升级

（1）卸载旧cuda

卸载原来的cuda8.0（注意：不需要卸载显卡驱动，不要给自己找麻烦）：

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl

卸载之后，会发现 /usr/local/cuda-8.0目录下任然有文件存在，这是cudnn文件，所以还需要将cuda-8.0文件删除干净：

sudo rm -rf /usr/local/cuda-8.0

（2）安装cuda9.0

下载cuda9.0的安装文件 cuda_9.0.176_384.81_linux.run（点击进入下载界面）

执行如下命令：

sudo sh cuda_9.0.176_384.81_linux.run

然后用一本书的书角压住回车键，直到服务条款显示到100%。接着按下面的步骤选择：

accept

n（不要安装driver）

y

y

y

安装完成后，设置环境变量。

（3）修改环境变量

打开主目录下的 .bashrc文件，将相关路径按下面的格式进行修改。例如我的.bashrc文件在/home/wangyuanwei下，如果没有找到，则按Ctrl+H键显示隐藏文件。

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

在终端运行：source ~/.bashrc

（4）检查当前的cuda版本

在终端输入nvcc --version，如果显示下面的文子就说明安装成功了。

2.cudnn的升级

cudnn的升级方法与cuda的升级方法类似。

（1）删除旧版本的cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn

（2）安装新版本的cudnn

下载所需要的cudnn版本（点击进入下载界面）

解压文件，并在解压出的cuda文件夹中打开终端，执行如下命令：

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

（3）建立软链接

在/usr/local/cuda/lib64目录下打开终端，执行如下指令：

sudo chmod +r libcudnn.so.7.0.5
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so     
sudo ldconfig

（4）查看当前cudnn的版本

在终端输入

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

如果出现下图所示版本信息（三个数字的意思是当前版本为7.0.5），说明安装成功。