Google Colab cudnn RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

I am getting this error while running trained model on google colab.

I tried to search this error that how to solve but I didn’t get any answer. Please tell me how to solve this error. Thank you in advance.

Could you post a minimal and executable code snippet to reproduce the issue and post also the output of python -m torch.utils.collect_env, please?

Just as a general advice on colab runtime, try to reproduce your error several times by deleting and re-initializing runtime and re-running code. It can be that the specific runtime at the moment might be problematic. This happened to me a few times

Edit: This is of course unless you can spot the origin of the problem