Ubuntu16.04 CUDA CUDNN卸载安装升级
最近需要构建tensorflow-gpu1.7.0的环境,由于原先使用的是老版本tensorflow,cuda版本为8.0,cudnn版本为5.1,所以需要进行cuda以及cudnn的升级,在此记录一下。
cuda的卸载与安装
(1)旧版本cuda卸载
我们这里需要卸载原来的cuda8.0,在此我没有去卸载显卡驱动,我尝试安装cuda9.2的版本,发现cuda9.2需要392+的版本才能匹配,所以还是老老实实用cuda9.0:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
上一步只会卸载cuda相关,原先添加的cudnn相关文件依旧保留在cuda-8.0文件夹下,接下来一步会完全移除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)修改环境变量
完成到上一步,cuda安装已经差不多了,接下来就要为cuda添加环境变量了.
打开主目录下的 .bashrc文件:
gedit ~/.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,使更改生效:
source ~/.bashrc
(4)查看当前cuda版本号
在terminal中入:
nvcc --version
显示cuda版本信息如下表明安装成功啦~~~
cudnn的卸载与安装
(1)删除旧版本cudnn
terminal中输入:
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn
(2)安装新版本cudnn
根据需要下载对应的cudnn版本,这里需要你的Nvidia帐号,不然老黄不给你玩儿(点击进入下载界面)
直接下载Linux binary版本,后面解压要用
下载结束后,我们解压下载包,解压出来的文件夹名称会变成“cuda”
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
(3)建立cudnn软连接
先切换目录到/usr/local/cuda/lib64
cd /usr/local/cuda/lib64
terminal输入如下命令建立软连接,不要忘记ldconfig:
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
显示如下图表明cudnn7.5.0安装成功:
至此cuda,cudnn的升级就搞定啦,另外cuda可以多版本并存,后面再更新~