Failed to run tensorflow pix2pix example

Does anyone encounter similar issues?
When running pix2pix example from tensorflow pix2pix eaxmple in WSL2, I got

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]

To reproduce the error, git clone the jupyter notebook https://github.com/tensorflow/docs/blob/master/site/en/tutorials/generative/pix2pix.ipynb

and run it in WSL2 cuda. I’ve tried in various settings on my laptop with RTX2070,

In all this settings, https://github.com/tensorflow/docs/blob/master/site/en/tutorials/keras/classification.ipynb
works fine
and the pix2pix example works in the windows host tensorflow.

Thanks for reaching out,

From your description I don’t think there is any issue with your setup. We will try internally and see if we can reproduce that issue. We will keep you posted,

Thanks again !