GPU训练完后自动关机：
在代码最末尾添加

import os
os.system("shutdown")