diff --git a/tox.ini b/tox.ini
index 0794c417..777ed4ce 100644
--- a/tox.ini
+++ b/tox.ini
@@ -10,6 +10,9 @@ package = wheel
 wheel_build_env = pkg
 deps =
     pytest
+setenv =
+    PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu128
+    UV_INDEX=https://download.pytorch.org/whl/cu128
 install_command = pip install \
                   -c constraints-dev.txt \
                   {opts} {packages}