Update ONNX script

lutzroeder · Nov 22, 2024 · 1693ee4 · 1693ee4
1 parent 9fdd6c1
commit 1693ee4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
 }