初始化换发型项目:3个微服务代码 + 部署脚本
包含: - hair_service_sd: 换发型/换发色算法服务 (端口 8801) - photo_service: LoRA 训练调度服务 (端口 32678) - stable-diffusion-webui: SD WebUI 推理服务 (端口 57860) - kohya_ss_home: 训练环境代码 - meidaojia: 监控测试脚本 - setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查) - start_all_services.sh: 启动3个服务 - configure.ini.template: 路径模板化 (BASE_DIR自动推导) - conda_envs/py310.yml: py310 环境定义 大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除, 由网盘单独上传。
This commit is contained in:
Executable
+69
@@ -0,0 +1,69 @@
|
||||
#!/bin/bash
|
||||
|
||||
# 定义连接参数
|
||||
HOST="northwest1.gpugeek.com"
|
||||
PORT="55139"
|
||||
USER="root"
|
||||
PASSWORD="uaUMSzmAHyka2ZcBvQHZ5My9PUNRP78T"
|
||||
TARGET_DIR="/root/project/hair_service_sd"
|
||||
COMMAND="if ! pgrep -f 'run_copy_cost_colorb64.py'; then bash start_services.sh; fi"
|
||||
|
||||
# 使用expect工具自动化交互过程
|
||||
/usr/bin/expect <<EOF
|
||||
set timeout 30
|
||||
|
||||
# 启动ssh连接
|
||||
spawn ssh -p $PORT $USER@$HOST
|
||||
|
||||
expect {
|
||||
# 处理首次连接的确认提示
|
||||
"you sure you want to continue connecting" {
|
||||
send "yes\r"
|
||||
exp_continue
|
||||
}
|
||||
# 匹配各种可能的密码提示形式
|
||||
"*password:" {
|
||||
send "$PASSWORD\r"
|
||||
}
|
||||
"Password:" {
|
||||
send "$PASSWORD\r"
|
||||
}
|
||||
"*assword:" {
|
||||
send "$PASSWORD\r"
|
||||
}
|
||||
}
|
||||
|
||||
# 等待登录成功提示
|
||||
expect {
|
||||
"*\$ " {}
|
||||
"*# " {}
|
||||
}
|
||||
|
||||
# 进入目标目录
|
||||
send "cd $TARGET_DIR\r"
|
||||
expect {
|
||||
"*\$ " {}
|
||||
"*# " {}
|
||||
}
|
||||
|
||||
# 执行检查进程并启动服务的命令
|
||||
send "$COMMAND\r"
|
||||
|
||||
# 等待命令执行完成
|
||||
expect {
|
||||
"*\$ " {
|
||||
send "exit\r"
|
||||
}
|
||||
"*# " {
|
||||
send "exit\r"
|
||||
}
|
||||
timeout {
|
||||
send "exit\r"
|
||||
}
|
||||
}
|
||||
|
||||
# 等待ssh会话结束
|
||||
expect eof
|
||||
EOF
|
||||
|
||||
echo "自动化任务已完成"
|
||||
Reference in New Issue
Block a user