From d8b3972bdcea7e6e6985b5fbb1def4f44b19b453 Mon Sep 17 00:00:00 2001
From: Ethanfel <ethan.fel@ts-pc.fr>
Date: Sun, 19 Apr 2026 16:21:36 +0200
Subject: [PATCH] fix: ensure setup scripts use correct PyTorch index for
 transitive deps

pip install -r requirements.txt can pull CPU-only torchvision via
transitive dependencies (timm, ultralytics). Adding --extra-index-url
with the CUDA wheel index ensures all torch packages stay on the
correct build. Applied to both Linux and Windows setup scripts.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 setup-windows.ps1 | 2 +-
 setup_env.sh      | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup-windows.ps1 b/setup-windows.ps1
index 7cf5679..61a6664 100644
--- a/setup-windows.ps1
+++ b/setup-windows.ps1
@@ -38,7 +38,7 @@ pip install torch torchaudio torchvision --index-url $torchIndex
 
 # ── Python deps ───────────────────────────────────────────
 Write-Host "`nInstalling project dependencies..."
-pip install -r (Join-Path $root "requirements.txt")
+pip install -r (Join-Path $root "requirements.txt") --extra-index-url $torchIndex
 
 # ── libmpv ────────────────────────────────────────────────
 $mpvDll = Join-Path $root "libmpv-2.dll"
diff --git a/setup_env.sh b/setup_env.sh
index 89dfff5..28600e3 100755
--- a/setup_env.sh
+++ b/setup_env.sh
@@ -69,7 +69,7 @@ setup_conda() {
     pip install torch torchaudio torchvision --index-url "$TORCH_INDEX"
 
     echo "  Installing project dependencies..."
-    pip install -r "$SCRIPT_DIR/requirements.txt"
+    pip install -r "$SCRIPT_DIR/requirements.txt" --extra-index-url "$TORCH_INDEX"
 
     echo ""
     echo "Done! Activate with:"
@@ -94,7 +94,7 @@ setup_venv() {
     pip install torch torchaudio torchvision --index-url "$TORCH_INDEX"
 
     echo "  Installing project dependencies..."
-    pip install -r "$SCRIPT_DIR/requirements.txt"
+    pip install -r "$SCRIPT_DIR/requirements.txt" --extra-index-url "$TORCH_INDEX"
 
     echo ""
     echo "Done! Activate with:"