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,
- WSL2, CUDA, Ubuntu 20.04, tensorflow 2.2.0, cuda 10.1.243-3(apt installed) and cudnn 7.6.5 in /usr/lib/x86_64-linux-gnu/
- Same as above but tf-nightly
- nvcr.io/nvidia/tensorflow:20.03-tf2-py3
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.