apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(build): pin Torch to 1.x for now

This commit is contained in:
Sean Sube 2023-03-17 21:31:15 -05:00
parent 4de8c160c9
commit a21750ff8e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
5 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.4
torch torch>=1.12,<2
torchvision torchvision
onnxruntime onnxruntime>=1.13

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

@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch>=1.12,<2
torchvision torchvision
--extra-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

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

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch>=1.12,<2
torchvision torchvision
onnxruntime-directml onnxruntime-directml

2
api/requirements/cpu.txt
View File

@ -1,3 +1,3 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch torch>=1.12,<2
torchvision torchvision

4
api/requirements/nvidia.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://download.pytorch.org/whl/cu117
torch torch>=1.12,<2
torchvision torchvision
onnxruntime-gpu onnxruntime-gpu>=1.13