1.cuda的升级
(1)卸载旧cuda
卸载原来的cuda8.0(注意:不需要卸载显卡驱动,不要给自己找麻烦):
/usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
卸载之后,会发现 /usr/local/cuda-8.0目录下任然有文件存在,这是cudnn文件,所以还需要将cuda-8.0文件删除干净:
rm -rf /usr/local/cuda-8.0
(2)安装cuda9.0
下载cuda9.0的安装文件 cuda_9.0.176_384.81_linux.run
执行如下命令:
sh cuda_9.0.176_384.81_linux.run
然后用一本书的书角压住回车键,直到服务条款显示到100%。接着按下面的步骤选择:
accept
n(不要安装driver)
y
y
y
安装完成后,设置环境变量。
(3)修改环境变量
打开主目录下的 .bashrc文件,将相关路径按下面的格式进行修改。
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
rm -rf /usr/local/cuda/include/cudnn.h
rm -rf /usr/local/cuda/lib64/libcudnn
(2)安装新版本的cudnn
下载所需要的cudnn版本
解压文件,并在解压出的cuda文件夹中打开终端,执行如下命令:
cp include/cudnn.h /usr/local/cuda/include/
cp lib64/lib* /usr/local/cuda/lib64/
(3)建立软链接
在/usr/local/cuda/lib64目录下打开终端,执行如下指令:
chmod +r libcudnn.so.7.0.5
ln -sf libcudnn.so.7.0.5 libcudnn.so.7
ln -sf libcudnn.so.7 libcudnn.so
ldconfig
(4)查看当前cudnn的版本
在终端输入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现下图所示版本信息(三个数字的意思是当前版本为7.0.5),说明安装成功。