apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): update requirements for torch 2.0

This commit is contained in:
Sean Sube 2023-04-09 17:19:40 -05:00
parent 34f1973707
commit 89ebbb8536
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
8 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/requirements/amd-linux-nightly.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.2
torch==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
https://download.onnxruntime.ai/onnxruntime_training-1.15.0.dev20230326001%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl https://download.onnxruntime.ai/onnxruntime_training-1.15.0.dev20230326001%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

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.2
torch==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
https://download.onnxruntime.ai/onnxruntime_training-1.14.1%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl https://download.onnxruntime.ai/onnxruntime_training-1.14.1%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

4
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==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
--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

4
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==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
onnxruntime-directml>=1.13 onnxruntime-directml>=1.13

4
api/requirements/cpu-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==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
--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 ort-nightly

4
api/requirements/cpu.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==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
onnxruntime>=1.13 onnxruntime>=1.13

4
api/requirements/nvidia-nightly.txt
View File

@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
--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-gpu ort-nightly-gpu

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==1.13.1 torch==2.0.0
torchvision==0.14.1 torchvision==0.15.1
onnxruntime-gpu>=1.13 onnxruntime-gpu>=1.13