diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index aecfb7deb..f99126750 100644 --- a/api/requirements/amd-linux.txt +++ b/api/requirements/amd-linux.txt @@ -1,5 +1,4 @@ ---index-url https://download.pytorch.org/whl/rocm5.2 +--extra-index-url https://download.pytorch.org/whl/rocm5.2 torch ---index-url https://download.pytorch.org/whl/rocm5.2 torchvision onnxruntime \ No newline at end of file diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt index 2d8225ca8..abebd84fe 100644 --- a/api/requirements/amd-windows-nightly.txt +++ b/api/requirements/amd-windows-nightly.txt @@ -1,6 +1,5 @@ ---index-url https://download.pytorch.org/whl/cpu +--extra-index-url https://download.pytorch.org/whl/cpu torch ---index-url https://download.pytorch.org/whl/cpu torchvision ---index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ +--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 681b884a8..295fef7f8 100644 --- a/api/requirements/amd-windows.txt +++ b/api/requirements/amd-windows.txt @@ -1,5 +1,4 @@ ---index-url https://download.pytorch.org/whl/cpu +--extra-index-url https://download.pytorch.org/whl/cpu torch ---index-url https://download.pytorch.org/whl/cpu torchvision onnxruntime-directml \ No newline at end of file diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt index 4de3bff59..3e046e928 100644 --- a/api/requirements/cpu.txt +++ b/api/requirements/cpu.txt @@ -1,4 +1,3 @@ ---index-url https://download.pytorch.org/whl/cpu +--extra-index-url https://download.pytorch.org/whl/cpu torch ---index-url https://download.pytorch.org/whl/cpu torchvision diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt index fcca051de..eb1dae62c 100644 --- a/api/requirements/nvidia.txt +++ b/api/requirements/nvidia.txt @@ -1,5 +1,4 @@ ---index-url https://download.pytorch.org/whl/cu117 +--extra-index-url https://download.pytorch.org/whl/cu117 torch ---index-url https://download.pytorch.org/whl/cu117 torchvision onnxruntime-gpu