Clean state

pipeline/scripts/bootstrap_env.sh

@@ -0,0 +1,48 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
+cd "$ROOT_DIR"
+
+PYTHON_BIN="${PYTHON_BIN:-python3}"
+USE_SYSTEM_SITE_PACKAGES="${USE_SYSTEM_SITE_PACKAGES:-1}"
+SKIP_TORCH_INSTALL="${SKIP_TORCH_INSTALL:-1}"
+
+VENV_ARGS=()
+if [ "$USE_SYSTEM_SITE_PACKAGES" = "1" ]; then
+    VENV_ARGS+=(--system-site-packages)
+fi
+
+"$PYTHON_BIN" -m venv "${VENV_ARGS[@]}" .venv
+source .venv/bin/activate
+
+python -m pip install --upgrade pip setuptools wheel
+
+# Capture system torch versions before venv installs override them
+TORCH_VERSION="$($PYTHON_BIN -c 'import torch; print(torch.__version__)' 2>/dev/null || true)"
+TV_VERSION="$($PYTHON_BIN -c 'import torchvision; print(torchvision.__version__)' 2>/dev/null || true)"
+
+if [ "$SKIP_TORCH_INSTALL" = "1" ]; then
+    filtered_requirements="$(mktemp)"
+    grep -Ev '^(torch|torchvision)([<>=!~].*)?$' requirements.txt > "$filtered_requirements"
+    
+    # Pin system torch/torchvision so transitive deps (e.g. facenet-pytorch) can't downgrade them
+    constraints_file="$(mktemp)"
+    [ -n "$TORCH_VERSION" ] && echo "torch==$TORCH_VERSION" >> "$constraints_file"
+    [ -n "$TV_VERSION" ] && echo "torchvision==$TV_VERSION" >> "$constraints_file"
+    python -m pip install -c "$constraints_file" -r "$filtered_requirements"
+    rm -f "$filtered_requirements" "$constraints_file"
+else
+    python -m pip install -r requirements.txt
+fi
+
+mkdir -p classifier/outputs/logs classifier/outputs/models classifier/outputs/analysis classifier/outputs/figures classifier/outputs/pipeline
+
+python - <<'PY'
+try:
+    import torch
+    print(f"torch={torch.__version__} cuda_available={torch.cuda.is_available()}")
+except Exception as exc:
+    print(f"torch check failed: {exc}")
+PY
+