Google Colab RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

Which PyTorch, CUDA, and cudnn version are you using?