Error : Failed to get convolution algorithm.

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)