1. 修改了一些bug
2. 做了一些调整,比如启动脚本,支持了tenmsorboard
This commit is contained in:
49
start_all.sh
49
start_all.sh
@@ -1,6 +1,6 @@
|
||||
#!/bin/bash
|
||||
# YG_FT_Base 统一启动脚本
|
||||
# 同时启动后端 API 服务和 Web 静态服务器
|
||||
# 同时启动后端 API 服务、Web 静态服务器和 TensorBoard
|
||||
# 使用方法: bash start_all.sh
|
||||
|
||||
# 自动修复脚本换行符
|
||||
@@ -34,6 +34,7 @@ fi
|
||||
echo "📦 端口配置:"
|
||||
echo " - 后端 API: $API_PORT"
|
||||
echo " - Web 服务器: $WEB_PORT"
|
||||
echo " - TensorBoard: 6006"
|
||||
echo ""
|
||||
|
||||
# 检查端口是否已被占用
|
||||
@@ -70,6 +71,36 @@ start_api() {
|
||||
echo "$API_PID" > /tmp/ygft_api.pid
|
||||
}
|
||||
|
||||
# 启动 TensorBoard 服务
|
||||
start_tensorboard() {
|
||||
echo ""
|
||||
echo "🚀 启动 TensorBoard 服务..."
|
||||
|
||||
# 检查端口
|
||||
if ! check_port 6006; then
|
||||
echo "⚠️ 端口 6006 已被占用,TensorBoard 可能已在运行"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 确保日志目录存在
|
||||
LOG_DIR="/app/base/saves"
|
||||
if [ ! -d "$LOG_DIR" ]; then
|
||||
LOG_DIR="$SCRIPT_DIR/saves"
|
||||
fi
|
||||
|
||||
if [ ! -d "$LOG_DIR" ]; then
|
||||
echo "⚠️ 日志目录不存在,跳过 TensorBoard 启动"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 启动 TensorBoard(后台运行)
|
||||
nohup tensorboard --logdir "$LOG_DIR" --port 6006 --bind_all > "$LOG_DIR/tensorboard.log" 2>&1 &
|
||||
TB_PID=$!
|
||||
echo "✅ TensorBoard 服务已启动 (PID: $TB_PID, 端口: 6006)"
|
||||
echo "$TB_PID" > /tmp/ygft_tensorboard.pid
|
||||
echo "📊 TensorBoard 访问地址: http://localhost:6006"
|
||||
}
|
||||
|
||||
# 启动 Web 静态服务器
|
||||
start_web() {
|
||||
echo ""
|
||||
@@ -105,9 +136,16 @@ stop_all() {
|
||||
echo "✅ Web 服务已停止"
|
||||
fi
|
||||
|
||||
if [ -f /tmp/ygft_tensorboard.pid ]; then
|
||||
kill $(cat /tmp/ygft_tensorboard.pid) 2>/dev/null
|
||||
rm /tmp/ygft_tensorboard.pid
|
||||
echo "✅ TensorBoard 服务已停止"
|
||||
fi
|
||||
|
||||
# 清理可能残留的进程
|
||||
pkill -f "src/main.py" 2>/dev/null
|
||||
pkill -f "http.server $WEB_PORT" 2>/dev/null
|
||||
pkill -f "tensorboard.*6006" 2>/dev/null
|
||||
}
|
||||
|
||||
# 显示状态
|
||||
@@ -128,16 +166,24 @@ status() {
|
||||
echo "❌ Web 服务: 未运行"
|
||||
fi
|
||||
|
||||
if [ -f /tmp/ygft_tensorboard.pid ] && kill -0 $(cat /tmp/ygft_tensorboard.pid) 2>/dev/null; then
|
||||
echo "✅ TensorBoard: 运行中 (PID: $(cat /tmp/ygft_tensorboard.pid), 端口: 6006)"
|
||||
else
|
||||
echo "❌ TensorBoard: 未运行"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo "🌐 访问地址:"
|
||||
echo " - 后端 API: http://localhost:$API_PORT"
|
||||
echo " - Web 页面: http://localhost:$WEB_PORT/pages/main.html"
|
||||
echo " - TensorBoard: http://localhost:6006"
|
||||
}
|
||||
|
||||
# 主菜单
|
||||
case "$1" in
|
||||
start)
|
||||
start_api
|
||||
start_tensorboard
|
||||
start_web
|
||||
echo ""
|
||||
echo "===================================="
|
||||
@@ -153,6 +199,7 @@ case "$1" in
|
||||
stop_all
|
||||
sleep 1
|
||||
start_api
|
||||
start_tensorboard
|
||||
start_web
|
||||
echo ""
|
||||
echo "===================================="
|
||||
|
||||
Reference in New Issue
Block a user