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),说明安装成功。

你可能感兴趣的:(深度学习)