Skip to content

Commit

Permalink
set torch and ORT as extra indexes
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Mar 6, 2023
1 parent 05c628d commit ccd15c1
Show file tree
Hide file tree
Showing 5 changed files with 6 additions and 11 deletions.
3 changes: 1 addition & 2 deletions api/requirements/amd-linux.txt
Original file line number Diff line number Diff line change
@@ -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
5 changes: 2 additions & 3 deletions api/requirements/amd-windows-nightly.txt
Original file line number Diff line number Diff line change
@@ -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
3 changes: 1 addition & 2 deletions api/requirements/amd-windows.txt
Original file line number Diff line number Diff line change
@@ -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
3 changes: 1 addition & 2 deletions api/requirements/cpu.txt
Original file line number Diff line number Diff line change
@@ -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
3 changes: 1 addition & 2 deletions api/requirements/nvidia.txt
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit ccd15c1

Please sign in to comment.