KICCO_AI_IMAGE/start_flux_api.sh

#!/bin/bash

# 虚拟环境路径 - 请修改为您实际的虚拟环境路径
VENV_PATH="/zhxiao/project/KICCO_AI_IMAGE/venv"

# 激活虚拟环境
echo "正在激活Python虚拟环境..."
source "$VENV_PATH/bin/activate"

# 检查虚拟环境是否成功激活
if [ $? -ne 0 ]; then
    echo "错误: 无法激活虚拟环境，请检查路径: $VENV_PATH"
    exit 1
fi

echo "已激活虚拟环境: $(which python3)"

# 创建日志目录
mkdir -p logs

# 启动第一个实例使用GPU 0，端口8000
echo "启动 GPU 0 上的服务实例，端口8000..."
CUDA_VISIBLE_DEVICES=0 nohup python3 flux_style_shaper_api/flux_style_shaper_api.py --port 8001 > logs/flux_api_gpu0.log 2>&1 &
INSTANCE1_PID=$!
echo "GPU 0服务已启动，进程ID: $INSTANCE1_PID"

# 启动第二个实例使用GPU 1，端口8001
echo "启动 GPU 1 上的服务实例，端口8001..."
CUDA_VISIBLE_DEVICES=1 nohup python3 flux_style_shaper_api/flux_style_shaper_api.py --port 8002 > logs/flux_api_gpu1.log 2>&1 &
INSTANCE2_PID=$!
echo "GPU 1服务已启动，进程ID: $INSTANCE2_PID"

echo "所有服务实例已启动!"
echo "可以通过以下地址访问服务："
echo "  - 实例1: http://$(hostname -I | awk '{print $1}'):8001"
echo "  - 实例2: http://$(hostname -I | awk '{print $1}'):8002"
echo "查看日志："
echo "  tail -f logs/flux_api_gpu0.log"
echo "  tail -f logs/flux_api_gpu1.log"

# 保持虚拟环境激活状态
# deactivate