Did you check your GPU temperature? I met your problem these days.Finally I checked it out.
In terminal,
watch -n 0.5 nvidia-smi -q -i 0,1 -d TEMPERATURE

Then,start your train. If GPU Current Temp > GPU Shutdown Temp, The system will crash