diff --git a/x2paddle/convert.py b/x2paddle/convert.py index e80f05adfd73371a98125e35f8f1107bf8bf0d49..7010828ab09d8ee67d1599a9cc46a11e2ebfa15c 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -125,11 +125,11 @@ def caffe2paddle(proto, weight, save_dir, caffe_proto): s = sys.version if s.startswith('2'): import commands - pb_version = commands.getstatusoutput('protoc --version')[1] + pb_version = commands.getstatusoutput('pip list | grep protobuf')[1] else: import subprocess - pb_version = subprocess.getstatusoutput('protoc --version')[1] - ver_str = pb_version.split(' ')[-1].replace('.', '') + pb_version = subprocess.getstatusoutput('pip list | grep protobuf')[1] + ver_str = pb_version.replace(' ', '').split('protobuf')[-1].replace('.', '') ver_int = int(ver_str) assert ver_int >= 360, 'The version of protobuf must be larger than 3.6.0!' print("Now translating model from caffe to paddle.")