fix(buil): limit torchvision versions to match torch

ssube · Mar 18, 2023 · fc9245f · fc9245f
1 parent a21750f
commit fc9245f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm5.4
 torch>=1.12,<2
-torchvision
+torchvision>=0.13,<0.15
 onnxruntime>=1.13
diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch>=1.12,<2
-torchvision
+torchvision>=0.13,<0.15
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
 ort-nightly-directml
diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch>=1.12,<2
-torchvision
+torchvision>=0.13,<0.15
 onnxruntime-directml
diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt
@@ -1,3 +1,3 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch>=1.12,<2
-torchvision
+torchvision>=0.13,<0.15
diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu117
 torch>=1.12,<2
-torchvision
+torchvision>=0.13,<0.15
 onnxruntime-gpu>=1.13