Update ONNX versions (apache#8304)

zxy844288792 · Jan 13, 2022 · 8e46f9e · 8e46f9e
1 parent 791f26b
commit 8e46f9e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/docker/install/ubuntu_install_onnx.sh b/docker/install/ubuntu_install_onnx.sh
@@ -20,9 +20,10 @@ set -e
 set -u
 set -o pipefail
 
-# fix to certain version for now
-pip3 install onnx==1.6.0
-pip3 install onnxruntime==1.0.0
+# We need to fix the onnx version because changing versions tends to break tests
+# TODO(mbrookhart): periodically update
+pip3 install onnx==1.8.1
+pip3 install onnxruntime==1.7.0
 
 # torch depends on a number of other packages, but unhelpfully, does
 # not expose that in the wheel!!!