After training a PyTorch model on a GPU for several hours, the program fails with the error RuntimeError: cuDNN error: …