From a21750ff8eabbf2b756e7c0a52301a4618c8adf3 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Mar 2023 21:31:15 -0500 Subject: [PATCH] fix(build): pin Torch to 1.x for now --- api/requirements/amd-linux.txt | 6 +++--- api/requirements/amd-windows-nightly.txt | 2 +- api/requirements/amd-windows.txt | 2 +- api/requirements/cpu.txt | 2 +- api/requirements/nvidia.txt | 4 ++-- 5 files changed, 8 insertions(+), 8 deletions(-) diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index f9912675..d8f9f60e 100644 --- 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 \ No newline at end of file +onnxruntime>=1.13 \ No newline at end of file diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt index abebd84f..bbf28294 100644 --- 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 \ No newline at end of file diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt index 295fef7f..86189e90 100644 --- 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 \ No newline at end of file diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt index 3e046e92..bbbb77d8 100644 --- 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 diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt index eb1dae62..1ff079fb 100644 --- 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