Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决办法

配置:
Tensorflow-gpu 2.0 alpha
CUDA 10.0 CUDNN7.5
WINDOWS 10
Python 3.6

刚开始跑一般的神经网络出现了个这么个问题,直接用下面链接的方法

https://blog.csdn.net/weixin_45783552/article/details/103025198?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1

即在开头加入下面几句代码

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)

如果你是1.X的版本
有各种其他朋友写的方法,很好用

另外可能原因:
如果你是Jupyter notebook写的
更大的可能性是你同时打开了好几个文件

由于自己后来跑卷积神经网络,它在计算卷积的时候又报了这么个错,但之前的解决方法行不通

一种方法:
import os

os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘/gpu:0’

直接不用GPU跑,省事

自己的解决方法
升级了一下CUDNN版本,升成了7.6.1,再加上了前面的那段代码,完事

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值