Could you rerun the code via:
CUDA_LAUNCH_BLOCKING=1 python script.py args

and post the complete stack trace here?
Alternatively, you could also run the script on the CPU, which might yield a better error message.