fix(build): pin Torch to 1.x for now

2023-03-17 21:31:15 -05:00 · 2023-03-17 21:31:15 -05:00 · a21750ff8e
parent 4de8c160c9
commit a21750ff8e
5 changed files with 8 additions and 8 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.2
-torch
+--extra-index-url https://download.pytorch.org/whl/rocm5.4
+torch>=1.12,<2
 torchvision
-onnxruntime
+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
+torch>=1.12,<2
 torchvision
 --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
+torch>=1.12,<2
 torchvision
 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
+torch>=1.12,<2
 torchvision
--- a/api/requirements/nvidia.txt
+++ b/api/requirements/nvidia.txt
@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu117
-torch
+torch>=1.12,<2
 torchvision
-onnxruntime-gpu
+onnxruntime-gpu>=1.13