diff --git a/caffe2fluid/kaffe/caffe/resolver.py b/caffe2fluid/kaffe/caffe/resolver.py index e67be41f6455b1ef43c9ecf209ac4f0d220c512c..b237f23f8f149494a6ad2a3b4fc46c95a270109c 100644 --- a/caffe2fluid/kaffe/caffe/resolver.py +++ b/caffe2fluid/kaffe/caffe/resolver.py @@ -10,10 +10,16 @@ def import_caffepb(): p = os.path.dirname(p) p = os.path.join(p, '../../proto') sys.path.insert(0, p) - pb_version = subprocess.getstatusoutput('protoc --version')[1] + s = sys.version + if s.startswith('2'): + import commands + pb_version = commands.getstatusoutput('protoc --version')[1] + else: + import subprocess + pb_version = subprocess.getstatusoutput('protoc --version')[1] ver_str = pb_version.split(' ')[-1].replace('.', '') ver_int = int(ver_str) - assert vaer_int >= 360, 'The version of protobuf must be larger than 3.6.0!' + assert ver_int >= 360, 'The version of protobuf must be larger than 3.6.0!' import caffe_pb2 return caffe_pb2