32 lines
987 B
Python
32 lines
987 B
Python
import paramiko,time,sys
|
|
|
|
def main_optimize_model(pid):
|
|
try:
|
|
# 创建SSH对象
|
|
ssh = paramiko.SSHClient()
|
|
# 允许连接不在known_hosts文件中的主机
|
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
print('start connect')
|
|
# 连接服务器
|
|
ssh.connect('connect.bjc1.seetacloud.com', username='root', password='QoQA8q3Ds2VB', port=33733)
|
|
print('connect success')
|
|
|
|
# 使用nohup执行命令，并将输出重定向到文件
|
|
command = f'nohup seg_python /data/code/optimize_model_xj/optimize_model.py -pid {pid} >{pid}.log 2>&1&'
|
|
stdin, stdout, stderr = ssh.exec_command(command)
|
|
|
|
# 打印命令是否启动成功
|
|
print("Command executed")
|
|
finally:
|
|
# 确保连接被关闭
|
|
if ssh:
|
|
time.sleep(1)
|
|
ssh.close()
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if len(sys.argv) == 2:
|
|
pid = sys.argv[1]
|
|
main_optimize_model(pid)
|
|
|