Solved by ptrblck in post #4 
              
              
                You could try to set this environment argument via os.environ['CUDA_LAUNCH_BLOCKING'] = "1". 
This would have to be added before any other imports in your script to work properly.