Thanks for the update. Is the original (unmodified) Tacotron2 code running on this device?
If so, could you rerun your fork with CUDA_LAUNCH_BLOCKING=1 python train.py --output_directory ... and post the stack trace here again?