“Failed to get convolution algorithm......”卷积神经运行报错

最新推荐文章于 2022-06-01 14:46:18 发布

lzfshub

最新推荐文章于 2022-06-01 14:46:18 发布

阅读量389

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/weixin_40511249/article/details/108463524

本文详细介绍了在运行卷积神经网络时遇到cuDNN无法初始化的问题，并提供了一个有效的解决方案：通过TensorFlow设置GPU内存自动分配，避免了因内存分配不当导致的cuDNN初始化失败。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在运行卷积神经网络时报错

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无法初始化，所以请尝试查看上面是否打印了警告日志消息。）

解决办法

在程序的开头加上自动分配GPU内存的程序即可。亲测有效！

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)