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

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

In all this settings,
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 !