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

1143 0 0

作者:BooTurbo

使用训练环境：

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)

内容来源于网络如有侵权请私信删除

文章来源: 博客园

原文链接: https://www.cnblogs.com/booturbo/p/13463342.html

标签： AI 人工智能

你还没有登录，请先登录或注册！

还没有人评论，欢迎说说您的想法！

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

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

相关课程

热门标签

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

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

相关课程

热门标签

推荐文章

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

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