Specify onnxruntime version to install for CI tests (#23517)

Summary: No real change on the CI since currently the default latest is 0.4.0. houseroad bddppq Pull Request resolved: https://github.com/pytorch/pytorch/pull/23517 Differential Revision: D16550375 Pulled By: bddppq fbshipit-source-id: a669b8af678c79c4d6909300b28458fe6b7cd30c

Specify onnxruntime version to install for CI tests (#23517)
Summary: No real change on the CI since currently the default latest is 0.4.0. houseroad bddppq Pull Request resolved: https://github.com/pytorch/pytorch/pull/23517 Differential Revision: D16550375 Pulled By: bddppq fbshipit-source-id: a669b8af678c79c4d6909300b28458fe6b7cd30c
e1f89859 · BowenBao · Facebook Github Bot · c779eff5 · e1f89859
--- a/.jenkins/caffe2/test.sh
+++ b/.jenkins/caffe2/test.sh
@@ -130,7 +130,7 @@ if [[ "$BUILD_ENVIRONMENT" == *onnx* ]]; then
  # JIT C++ extensions require ninja, so put it into PATH.
  export PATH="/var/lib/jenkins/.local/bin:$PATH"
  if [[ "$BUILD_ENVIRONMENT" == *py3* ]]; then
-    pip install -q --user onnxruntime
+    pip install -q --user onnxruntime==0.4.0
  fi
  "$ROOT_DIR/scripts/onnx/test.sh"
 fi