GPU训练完后自动关机

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

import os
os.system("shutdown")