Update ONNX versions (#8304)

apache · Jun 22, 2021 · 1c6f2bc · 1c6f2bc
1 parent 5c2836c
commit 1c6f2bc
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!!!