安装cudnn

网上搜到了一些安装cudnn的步骤,都是这样的:

1. 从官网下载cudnn

2. 解压

3. 将解压出来的lib64和include中所有的内容都拷贝到/usr/local/cuda的lib64和include中

 不行。

看了一下解压出来的内容,lib64里面包含了两个软链接,直接cp过去软链接会失效。

解决办法有两个:

1. 用 cp -d,可以复制链接关系

2. 把lib64的路径添加到环境变量

正确的安装步骤

1. 从官网下载cudnn

NVIDIA cuDNN | NVIDIA Developer

第一次下要注册一下

我的配置是tensorflow-gpu-1.15.0,cuda 10.0,下载对应版本的cudnn就好

查看合适的版本:

tensorflow、python、cuda的对应关系

2. 解压

tar vxf cudnn-10.0-linux-x64-v7.6.0.64.tgz

3. 复制lib64下的内容

这一步有两个解决办法

1)cp -d

sudo cp -r -d [解压出来的lib64文件夹/*] usr/local/cuda/lib64/

2)修改环境变量

解压后会生成一个名叫cuda的文件夹

我把这个文件夹重命名成了 cudnn_7.6.0,然后把它移动到了/usr/local/

所以解压出来的lib64 的路径是

/usr/local/cudnn_7.6.0/lib64

把上面这个路径添加到LD_LIBRARY_PATH中

4. 复制include下的内容

把解压出来的include中的内容复制到 /usr/local/cuda/include中

sudo cp /usr/local/cudnn_7.6.0/include/* usr/local/cuda/include/

5. 重置读写权限

cudnn_7.6.0/include/ 里面只有一个 cudnn.h

sudo chmod a+r /usr/local/cuda/include/cudnn.h
  • 8
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值