apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

set torch and ORT as extra indexes

This commit is contained in:
Sean Sube 2023-03-05 18:15:41 -06:00
parent 05c628d8a6
commit ccd15c1921
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
5 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/requirements/amd-linux.txt
View File

@ -1,5 +1,4 @@
--index-url https://download.pytorch.org/whl/rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.2
torch torch
--index-url https://download.pytorch.org/whl/rocm5.2
torchvision torchvision
onnxruntime onnxruntime

5
api/requirements/amd-windows-nightly.txt
View File

@ -1,6 +1,5 @@
--index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch
--index-url https://download.pytorch.org/whl/cpu
torchvision 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 ort-nightly-directml

3
api/requirements/amd-windows.txt
View File

@ -1,5 +1,4 @@
--index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch
--index-url https://download.pytorch.org/whl/cpu
torchvision torchvision
onnxruntime-directml onnxruntime-directml

3
api/requirements/cpu.txt
View File

@ -1,4 +1,3 @@
--index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch
--index-url https://download.pytorch.org/whl/cpu
torchvision torchvision

3
api/requirements/nvidia.txt
View File

@ -1,5 +1,4 @@
--index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://download.pytorch.org/whl/cu117
torch torch
--index-url https://download.pytorch.org/whl/cu117
torchvision torchvision
onnxruntime-gpu onnxruntime-gpu