Solved by KFrank in post #2 
              
              
                Hello Mr. Air! 

Your laptop is overheating and shutting itself down to prevent permanent 
damage. 
When your cpu and/or gpu is running full tilt – and pytorch tries to make 
it run as fast as it can – that creates a lot of heat.  (So pytorch is doing 
its job here!) 

Some things you can try: 
Make…