diff --git a/paddleslim/common/load_model.py b/paddleslim/common/load_model.py index e611628bc4013845e0dc8e17943338fc5d8ba1c7..f4f3be289500e8a309fd4e15a9a0ddc2842bc2a5 100644 --- a/paddleslim/common/load_model.py +++ b/paddleslim/common/load_model.py @@ -121,9 +121,16 @@ def load_onnx_model(model_path, model_path), 'Not found `{}`, please check model path.'.format( model_path) try: - pkg.require('x2paddle') + import x2paddle + version = x2paddle.__version__ + v0, v1, v2 = version.split('.') + version_sum = int(v0) * 100 + int(v1) * 10 + int(v2) + if version_sum < 139: + _logger.error( + "x2paddle>=1.3.9 is required, please use \"pip install x2paddle\"." + ) except: - os.system('python -m pip install -U x2paddle==1.3.9') + os.system('python -m pip install -U x2paddle') # check onnx installation and version try: pkg.require('onnx')