feat(api): update requirements for torch 2.0
This commit is contained in:
parent
34f1973707
commit
89ebbb8536
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
--extra-index-url https://download.pytorch.org/whl/rocm5.4.2
|
||||
torch==2.0.0
|
||||
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
|
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
--extra-index-url https://download.pytorch.org/whl/rocm5.4.2
|
||||
torch==2.0.0
|
||||
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
|
|
@ -1,5 +1,5 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
|
||||
ort-nightly-directml
|
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
onnxruntime-directml>=1.13
|
|
@ -1,5 +1,5 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
|
||||
ort-nightly
|
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
onnxruntime>=1.13
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cu117
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
|
||||
ort-nightly-gpu
|
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cu117
|
||||
torch==1.13.1
|
||||
torchvision==0.14.1
|
||||
torch==2.0.0
|
||||
torchvision==0.15.1
|
||||
onnxruntime-gpu>=1.13
|
||||
|
|
Loading…
Reference in New Issue