From 2ec570d9ca6452687b202a905c18c706591c0260 Mon Sep 17 00:00:00 2001 From: liuchongming Date: Tue, 9 Feb 2021 12:34:27 +0800 Subject: [PATCH] Lower onnxruntime version for tester. --- mindinsight/mindconverter/graph_based_converter/framework.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mindinsight/mindconverter/graph_based_converter/framework.py b/mindinsight/mindconverter/graph_based_converter/framework.py index 515548d2..19940870 100644 --- a/mindinsight/mindconverter/graph_based_converter/framework.py +++ b/mindinsight/mindconverter/graph_based_converter/framework.py @@ -42,8 +42,11 @@ def onnx_lib_version_satisfied(): onnx = import_module("onnx") ort = import_module("onnxruntime") optimizer = import_module("onnxoptimizer.version") + if not lib_version_satisfied(getattr(ort, "__version__"), ONNXRUNTIME_MIN_VER): + log_console.warning("onnxruntime's version should be greater than %s, however current version is %s.", + ONNXRUNTIME_MIN_VER, ort.__version__) + if not lib_version_satisfied(getattr(onnx, "__version__"), ONNX_MIN_VER) \ - or not lib_version_satisfied(getattr(ort, "__version__"), ONNXRUNTIME_MIN_VER) \ or not lib_version_satisfied(getattr(optimizer, "version"), ONNXOPTIMIZER_MIN_VER, ONNXOPTIMIZER_MAX_VER): return False return True