Solved by ptrblck in post #2 
              
              
                Is your code working fine on the CPU? 
If so, could you rerun the code with CUDA_LAUNCH_BLOCKING=1 python script.py args and post the stack trace here, please?