【TensorFlow2.0】This is probably because cuDNN failed to initialize.错误修正和cuDNN版本更新

在Ubuntu16.04+TensorFlow2.0+CUDA10.1+cuDNN7.5.0环境下训练模型时遇到'Failed to get convolution algorithm'错误。该问题可能由代码或cuDNN版本不匹配引起。通过检查CUDA版本,下载并安装适配的cuDNN7.6.5,卸载旧版本,然后安装新版本的libcudnn7和libcudnn7-dev,解决了问题。
摘要由CSDN通过智能技术生成

今天在环境Ubuntu16.04+TensorFlow2.0+CUDA10.1+cuDNN7.5.0中训练模型,本来是使用CPU计算的,但是后来数据量扩大后CPU已经不能满足需求了,于是便安装了TensorFlow2.0-gpu进行训练。

错误

本来已经挺简单的,因为之前在TensorFlow-gpu1.14+CUDA10.1+cuDNN7.5.0的环境下进行过训练,但是没想到会出现错误

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

看错误提示和查询,发现这个错误有可能是我代码的问题,也有可能是因为我的cuDNN版本不适配。一开始我以为是第一点,看了其他人的博客在代码前面加上一些配置但不管用,于是只能采用第二个方法。

更新cuDNN过程

首先先使用命令 nvcc -V (注意V是大写)确定自己的CUDA版本,然后进入cuDNN下载页面

没有账号的话应注册NVIDIA账号,登录后会看到以下页面

根据自己的CUDA版本下载对应的cuDNN,有两种安装方式

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值