Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try ...

最新推荐文章于 2022-03-29 16:58:35 发布

敲代码的乔帮主

最新推荐文章于 2022-03-29 16:58:35 发布

阅读量418

点赞数

分类专栏： Error-我的bug

本文链接：https://blog.csdn.net/Suyebiubiu/article/details/106882115

版权

Error-我的bug 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

报错内容：

tensorflow.python.framework.errors_impl.UnknownError: 
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. [Op:Conv2D]

Process finished with exit code 1

这个是GPU内存不足造成的，不必怀疑CUDA和CuDNN的版本问题。可以在代码前加上下面的代码，即可成功运行。

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

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