初始化换发型项目: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
+85
@@ -0,0 +1,85 @@
|
||||
#!/bin/bash
|
||||
|
||||
# 定义目标目录和检查的进程名
|
||||
TARGET_DIR="/root/project/hair_service_sd"
|
||||
PROCESS_NAME="run_copy_cost_colorb64.py"
|
||||
START_COMMAND="bash start_services.sh"
|
||||
|
||||
# 读取servers.csv文件并逐行处理
|
||||
while IFS=, read -r name ssh_cmd password _; do
|
||||
# 去除可能存在的空格和引号
|
||||
name=$(echo "$name" | xargs)
|
||||
ssh_cmd=$(echo "$ssh_cmd" | xargs)
|
||||
password=$(echo "$password" | xargs)
|
||||
|
||||
echo "..................................正在处理服务器: $name ........................................"
|
||||
|
||||
# 从ssh命令中提取端口和主机
|
||||
port=$(echo "$ssh_cmd" | grep -oP '\-p\s*\K\d+')
|
||||
host=$(echo "$ssh_cmd" | grep -oP 'root@\K[^\s]+')
|
||||
|
||||
# 使用expect工具自动化交互过程
|
||||
/usr/bin/expect <<EOF
|
||||
set timeout 30
|
||||
|
||||
# 启动ssh连接
|
||||
spawn $ssh_cmd
|
||||
|
||||
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 "if ! pgrep -f '$PROCESS_NAME'; then echo '启动服务...'; $START_COMMAND; else echo '进程已在运行,跳过启动'; fi\r"
|
||||
|
||||
# 等待命令执行完成
|
||||
expect {
|
||||
"*\$ " {
|
||||
send "exit\r"
|
||||
}
|
||||
"*# " {
|
||||
send "exit\r"
|
||||
}
|
||||
timeout {
|
||||
send "exit\r"
|
||||
}
|
||||
}
|
||||
|
||||
# 等待ssh会话结束
|
||||
expect eof
|
||||
EOF
|
||||
|
||||
echo "已完成服务器: $name"
|
||||
# 添加延时
|
||||
echo "--------------------------------------------- 等待 ----------------------------------------------"
|
||||
sleep 5
|
||||
done < servers_test.csv
|
||||
|
||||
echo "所有服务器处理完成"
|
||||
Reference in New Issue
Block a user