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

最新推荐文章于 2021-10-24 21:58:57 发布

原创最新推荐文章于 2021-10-24 21:58:57 发布 · 267 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #tensorflow #cudnn

常见问题专栏收录该内容

13 篇文章

订阅专栏

当遇到'Failed to get convolution algorithm'错误时，这通常是因为GPU内存不足或版本不匹配。对于TensorFlow-gpu 2.0.0，可以通过设置`allow_growth`来动态分配GPU内存。然而，对于2.3.1版本，应使用`CUDA_VISIBLE_DEVICES`环境变量指定GPU设备。此解决方案针对特定TensorFlow版本有效。

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

如果tensorflow-gpu的版本是2.0.0，那直接用下面这5句代码就可以解决

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

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

原因可能是：gpu内存不够，不是版本不匹配
注意：这种解决方法只限于tensorflow-gpu的版本是2.0.0，同样在tensorflow-gpu的版本是2.3.1的情况下测试了一下，这种方法不可行。

如果是2.3.1版本
可以用下面方式
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘/device:GPU:0’