ubuntu16.04更改CUDNN 6.0 版本

在 ubuntu16.04 上安装tensorflow-gpu发现没有对应版本。查看之后,我的电脑是cuda 8.0,cudnn 7版本

所以需要更换cudnn版本为6.0,才可以安装tensorflow-gpu

一、下载linux版的CUDNN 6.0

官网下载:https://developer.nvidia.com/cuDNN,需要邮箱注册。

二、安装

1)解压

到下载的目录下:

tar zxvf cudnn-8.0-linux-x64-v6.0.tgz

2) 删除原来的CUDNN文件

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

3)cd 到之前解压的cuda文件夹下

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

4)cd到/usr/local/cuda/lib64/文件夹下,建立软链接

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.6.0.21
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so   
sudo ldconfig 

5)查看版本信息

cuda版本

cat /usr/local/cuda/version.txt

cudnn版本

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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值