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