Solved by ptrblck in post #2 
              
              
                Could you rerun the code via: 
CUDA_LAUNCH_BLOCKING=1 python script.pt args

and post the complete stack trace here, please?