Browse Source

Lower onnxruntime version for tester.

tags/v1.2.0-rc1
liuchongming 4 years ago
parent
commit
2ec570d9ca
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindinsight/mindconverter/graph_based_converter/framework.py

+ 4
- 1
mindinsight/mindconverter/graph_based_converter/framework.py View File

@@ -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


Loading…
Cancel
Save