From 4660b337fb7f8bde0e3ed19bce9dcc79fb374db1 Mon Sep 17 00:00:00 2001
From: Lucas Armand <lucas@vast.ai>
Date: Mon, 30 Mar 2026 14:46:38 -0700
Subject: [PATCH] Check for USE_SYSTEM_PYTHON

---
 start_server.sh | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/start_server.sh b/start_server.sh
index 89f2c1e..19a63f7 100755
--- a/start_server.sh
+++ b/start_server.sh
@@ -112,8 +112,21 @@ if ! grep -q "VAST" /etc/environment; then
     fi
 fi
 
-if [ ! -d "$ENV_PATH" ]
-then
+if [ "${USE_SYSTEM_PYTHON:-}" = "true" ]; then
+    echo "Using system Python: $(which python3)"
+    if ! which uv > /dev/null 2>&1; then
+        if ! curl -LsSf https://astral.sh/uv/install.sh | sh; then
+            report_error_and_exit "Failed to install uv package manager"
+        fi
+        if [[ -f ~/.local/bin/env ]]; then
+            if ! source ~/.local/bin/env; then
+                report_error_and_exit "Failed to source uv environment"
+            fi
+        fi
+    fi
+    install_vastai_sdk
+    touch ~/.no_auto_tmux
+elif [ ! -d "$ENV_PATH" ]; then
     echo "setting up venv"
     if ! which uv; then
         if ! curl -LsSf https://astral.sh/uv/install.sh | sh; then