美文网首页ubuntu
Error : Failed to get convolutio

Error : Failed to get convolutio

作者: h小吴y | 来源:发表于2020-12-09 17:05 被阅读0次

Error : 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.

https://github.com/tensorflow/tensorflow/issues/24828

image
# 添加下面代码即可
from tensorflow.compat.v1import ConfigProto

from tensorflow.compat.v1import InteractiveSession

config = ConfigProto()

config.gpu_options.allow_growth =True

session = InteractiveSession(config=config)

相关文章

网友评论

    本文标题:Error : Failed to get convolutio

    本文链接:https://www.haomeiwen.com/subject/pqnhgktx.html