Tensorflow issue:OP_REQUIRES failed at conv_ops_fused_impl.h:697 : Not found: No algorithm worked!

AI & Data Science Deep Learning (Training & Inference) cuDNN
5

Adding the code below right after the imports solved the issue for me ! Thanks !

physical_devices = tf.config.list_physical_devices(‘GPU’)
tf.config.experimental.set_memory_growth(physical_devices[0], True)