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

 

 

 

使用训练环境：

Tensorflow 2.2.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)





posted @ 2020-08-09 16:47  BooTurbo  阅读(3352)  评论(1)  编辑  收藏  举报