Thanks for the information.
I let the script run for 2400 epochs on a P100, but still couldn’t reproduce this error.
Unfortunately, I don’t have access to a Windows machine with this GPU to rerun the code.
Could you just for the sake of debugging disable cudnn via torch.backends.cudnn.enabled = False and rerun the code again?