tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm出错

神经网络训练时出现 无法获取卷积算法的问题tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

使用训练环境:
Tensorflow 2.3.0,CUDA 10.1,cuDNN 7.6.5。
当实例化VGG16卷积基时,出现了无法获取卷积算法的问题,提示可能因为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)

注意:tensorflow2.2 bug较多,改为tensorflow2.3较好。。。。

Guess you like

Origin blog.csdn.net/weixin_44885180/article/details/114315900