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可以多版本并存,后面再更新~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

_learnWay

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值