fix torchvision version

zewenli98 · zewenli98 · commit 74faecd5d83b · 2025-01-15T12:26:58.000-08:00
diff --git a/py/requirements.txt b/py/requirements.txt
@@ -1,6 +1,8 @@
 numpy
 packaging
 pybind11==2.6.2
+--extra-index-url https://download.pytorch.org/whl/test/cu126
 torch>=2.7.0.dev,<2.8.0
-torchvision>=0.22.0.dev,<0.23.0
+torchvision>=0.20.0.dev,<0.23.0
+--extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
