RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
To solve my problem, I did
pip uninstall torch torchvision torchaudio
Then
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
According to
https://pytorch.org/get-started/locally/