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