Avoid tensorflow.python.framework.errors_impl.UnknownError

[Hephaest]

System information

• OS Platform: Windows 10 64-bit
• TensorFlow version: 2.3.1
• Python version: 3.8.6
• CUDA version: 10.1
• cuDNN version: 7.6.5.32

Describe the problem

After installation by using GPU, the console reports the error:

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.
	 [[node functional_1/false_conv_0_0/conv1d (defined at C:\Users\Hephaest\AppData\Local\Programs\Python\Python38\lib\site-packages\tslearn\shapelets.py:451) ]] [Op:__inference_train_function_1239]

How I fix the problem

For TensorFlow 1.x user, according to issue answer from Tensorflow community, please copy and paste the following code after import tensorflow as tf:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

For TensorFlow 2.x user, according to issue answer from Tensorflow community, please copy and paste the following code after import tensorflow as tf:

gpu_devices = tf.config.experimental.list_physical_devices('GPU')
for device in gpu_devices:
    tf.config.experimental.set_memory_growth(device, True)