fix(buil): limit torchvision versions to match torch

2023-03-17 22:42:37 -05:00 · 2023-03-17 22:42:37 -05:00 · fc9245f4ee
parent a21750ff8e
commit fc9245f4ee
5 changed files with 5 additions and 5 deletions
--- 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
--- 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
--- 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
--- 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
--- 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