Is your code running without cudnn? Set torch.backends.cudnn.enabled = False and try to run your code again.
If you see another error message, try to run your code using:
CUDA_LAUNCH_BLOCKING=1 python script.py

and post the stack trace here.