fix(build): pin Torch to 1.x for now
This commit is contained in:
parent
4de8c160c9
commit
a21750ff8e
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||||
torch
|
||||
--extra-index-url https://download.pytorch.org/whl/rocm5.4
|
||||
torch>=1.12,<2
|
||||
torchvision
|
||||
onnxruntime
|
||||
onnxruntime>=1.13
|
|
@ -1,5 +1,5 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch
|
||||
torch>=1.12,<2
|
||||
torchvision
|
||||
--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
|
||||
torch>=1.12,<2
|
||||
torchvision
|
||||
onnxruntime-directml
|
|
@ -1,3 +1,3 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch
|
||||
torch>=1.12,<2
|
||||
torchvision
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cu117
|
||||
torch
|
||||
torch>=1.12,<2
|
||||
torchvision
|
||||
onnxruntime-gpu
|
||||
onnxruntime-gpu>=1.13
|
||||
|
|
Loading…
Reference in New Issue