From 1693ee48735ea82c11b93446bbd7d6a2d7b1fcbe Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Thu, 21 Nov 2024 18:02:33 -0800 Subject: [PATCH] Update ONNX script --- tools/onnx | 3 ++- tools/pytorch | 2 +- tools/sklearn | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tools/onnx b/tools/onnx index 6d9aa11319..10b497141a 100755 --- a/tools/onnx +++ b/tools/onnx @@ -92,7 +92,8 @@ install() { ;; esac ${python} -m pip install --quiet "./third_party/source/onnx" - ${python} -m pip install --quiet --upgrade ort-nightly --extra-index https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ + ${python} -m pip install --quiet --upgrade packaging flatbuffers protobuf numpy sympy + ${python} -m pip install --quiet --upgrade onnxruntime --pre --extra-index https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ deactivate } diff --git a/tools/pytorch b/tools/pytorch index d10d5dce04..12d3e97a86 100755 --- a/tools/pytorch +++ b/tools/pytorch @@ -37,7 +37,7 @@ install() { echo "pytorch install" venv ${python} -m pip install --quiet --upgrade wheel - ${python} -m pip install --quiet --upgrade --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu + ${python} -m pip install --quiet --upgrade torch torchvision torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/cpu # ${python} -m pip install --quiet --upgrade torch-neuron --index-url https://pip.repos.neuron.amazonaws.com deactivate } diff --git a/tools/sklearn b/tools/sklearn index 9c332d49d6..2987bbd348 100755 --- a/tools/sklearn +++ b/tools/sklearn @@ -38,7 +38,7 @@ install() { echo "sklearn install" venv ${python} -m pip install --quiet scipy - ${python} -m pip install --quiet --pre --extra-index https://pypi.anaconda.org/scipy-wheels-nightly/simple scikit-learn + ${python} -m pip install --quiet scikit-learn --pre --extra-index https://pypi.anaconda.org/scipy-wheels-nightly/simple deactivate }