From fc9245f4ee01de070dc4811ca7b3728ad749867c Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Mar 2023 22:42:37 -0500 Subject: [PATCH] fix(buil): limit torchvision versions to match torch --- api/requirements/amd-linux.txt | 2 +- api/requirements/amd-windows-nightly.txt | 2 +- api/requirements/amd-windows.txt | 2 +- api/requirements/cpu.txt | 2 +- api/requirements/nvidia.txt | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index d8f9f60e..bc96a271 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.4 torch>=1.12,<2 -torchvision +torchvision>=0.13,<0.15 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 bbf28294..f03d1e4f 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>=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 \ No newline at end of file diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt index 86189e90..4dc90905 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>=1.12,<2 -torchvision +torchvision>=0.13,<0.15 onnxruntime-directml \ No newline at end of file diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt index bbbb77d8..2cfff5a3 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>=1.12,<2 -torchvision +torchvision>=0.13,<0.15 diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt index 1ff079fb..a830ca76 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>=1.12,<2 -torchvision +torchvision>=0.13,<0.15 onnxruntime-gpu>=1.13