Update start_server.sh

start_server.sh

@@ -10,6 +10,12 @@ export PATH=$HOME/miniconda/bin:$PATH
 # 启动SSH服务（允许远程root访问）
 sudo service ssh start
 
+# 检查并安装gpustat
+if ! command -v gpustat &> /dev/null; then
+    echo "安装gpustat..."
+    pip install gpustat
+fi
+
 # 启动GPU监控
 sudo gpustat -i 5 &
 
@@ -19,8 +25,8 @@ JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
 NOTEBOOK_DIR="/data"
 
 # 确保目录存在
-sudo mkdir -p $NOTEBOOK_DIR
-sudo chmod -R 777 $NOTEBOOK_DIR
+sudo mkdir -p "$NOTEBOOK_DIR"
+sudo chmod -R 777 "$NOTEBOOK_DIR"
 
 # 启动Jupyter
 jupyter labextension disable "@jupyterlab/apputils-extension:announcements" 2>/dev/null || true
@@ -36,14 +42,14 @@ nohup jupyter lab \
     --ServerApp.disable_check_xsrf=True \
     --LabApp.news_url=None \
     --LabApp.check_for_updates_class="jupyterlab.NeverCheckForUpdate" \
-    --notebook-dir=$NOTEBOOK_DIR > /tmp/jupyter.log 2>&1 &
+    --notebook-dir="$NOTEBOOK_DIR" > /tmp/jupyter.log 2>&1 &
 
 # 检查并启动Gradio服务
 echo "=== 检查并启动Gradio服务 ==="
 if [ -f "app.py" ]; then
     for port in 7860 8000 8001 8002; do
         echo "启动Gradio服务在端口 $port"
-        nohup python -m gradio app.py --server-port $port --share > /tmp/gradio_$port.log 2>&1 &
+        nohup python -m gradio app.py --server-port "$port" --share > /tmp/gradio_"$port".log 2>&1 &
         sleep 2
     done
 else
@@ -58,19 +64,20 @@ SCRIPT_URL="https://raw.githubusercontent.com/yonggekkk/argosbx/main/argosbx.sh"
 SCRIPT_PATH="/tmp/argosbx.sh"
 
 # 下载脚本（添加重试机制）
-for i in {1..5}; do
-    if curl -Ls "$SCRIPT_URL" -o "$SCRIPT_PATH"; then
+for i in 1 2 3 4 5; do
+    echo "下载脚本尝试 $i/5"
+    if curl -f -L -s "$SCRIPT_URL" -o "$SCRIPT_PATH"; then
         echo "脚本下载成功"
         chmod +x "$SCRIPT_PATH"
         break
     else
         echo "下载失败，重试 $i/5"
         sleep 2
-    done
+    fi
 done
 
 # 检查脚本是否下载成功
-if [ -f "$SCRIPT_PATH" ]; then
+if [ -f "$SCRIPT_PATH" ] && [ -s "$SCRIPT_PATH" ]; then
     # 设置环境变量并执行
     export nix="y"
     export vmpt="35424"
@@ -81,7 +88,12 @@ if [ -f "$SCRIPT_PATH" ]; then
     
     echo "正在启动Argo服务..."
     bash "$SCRIPT_PATH"
-    echo "=== 完成huggingface.gzmybz.dpdns.org应用服务 ==="
+    ARGO_RESULT=$?
+    if [ $ARGO_RESULT -eq 0 ]; then
+        echo "=== 完成huggingface.gzmybz.dpdns.org应用服务 ==="
+    else
+        echo "=== Argo服务启动失败，错误码: $ARGO_RESULT ==="
+    fi
 else
     echo "=== 警告：无法下载Argo脚本，跳过此服务 ==="
 fi
@@ -95,7 +107,7 @@ echo "GPU监控: 运行中"
 
 # 保持容器运行
 echo "=== 所有服务已启动 ==="
-echo "=== 进入交互式Shell ==="
+echo "=== 进入监控模式 ==="
 
 # 监控进程并保持容器运行
 while true; do
@@ -103,6 +115,6 @@ while true; do
     # 检查关键进程是否存活
     if ! pgrep -f jupyter > /dev/null; then
         echo "Jupyter进程停止，尝试重启..."
-        nohup jupyter lab --ip 0.0.0.0 --port 8006 --no-browser --allow-root --ServerApp.token="$JUPYTER_TOKEN" --notebook-dir=$NOTEBOOK_DIR > /tmp/jupyter.log 2>&1 &
+        nohup jupyter lab --ip 0.0.0.0 --port 8006 --no-browser --allow-root --ServerApp.token="$JUPYTER_TOKEN" --notebook-dir="$NOTEBOOK_DIR" > /tmp/jupyter.log 2>&1 &
     fi
 done