Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
本文探讨了在TensorFlow中遇到cuDNN初始化失败的问题,提供了详细的代码配置,包括使用ConfigProto和InteractiveSession来管理GPU资源,确保cuDNN能够成功初始化。
2246

被折叠的 条评论
为什么被折叠?



