跳到主要内容

实例内调用命令关机

在训练或推理过程中如果不确定自己代码需要执行多久结束,但是希望训练成功或推理结果完成后立刻关机来进行节约成本,这类场景可以通过/usr/local/bin/poweroff命令来完成。

1.下载poweroff命令到实例中

下载poweroff命令
curl -sSL -o /usr/local/bin/poweroff https://oss-high-sq01.cdsgss.com/public-static-prod-gpuzoom/installation/tools/poweroff
chmod +x /usr/local/bin/poweroff

2.代码执行后调用poweroff命令进行实例关机

请保存好程序的日志,自动关机后标准输出中的日志将不再可见

# 假设您的程序原执行命令为
python train.py

# 那么可以在您的程序后跟上poweroff命令
python train.py; /usr/local/bin/poweroff # 用 ; 拼接意味着前边的指令不管执行成功与否,都会执行poweroff命令
python train.py && /usr/local/bin/poweroff # 用 && 拼接表示前边的命令执行成功后才会执行poweroff,请根据自己的需要选择

3.在代码中嵌入poweroff命令

import os

if __name__ == "__main__":
# xxxxxx
os.system("/usr/local/bin/poweroff")