42 lines
1.4 KiB
Bash
42 lines
1.4 KiB
Bash
#!/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 |