您的位置:首页 > 大数据 > 人工智能

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

2020-05-31 18:46 841 查看

配置:
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,再加上了前面的那段代码,完事

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐