Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try loo

tensorflow/keras 引入Conv2D报错

tensorflow.python.framework.errors_impl.UnknownError: 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. [Op:Conv2D]

解决方案1

import os

os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'