diff --git a/caffe2fluid/kaffe/caffe/resolver.py b/caffe2fluid/kaffe/caffe/resolver.py index e67be41f6455b1ef43c9ecf209ac4f0d220c512c..4ff3a9d6acbef7f5732a8bdee6c1903b70392870 100644 --- a/caffe2fluid/kaffe/caffe/resolver.py +++ b/caffe2fluid/kaffe/caffe/resolver.py @@ -6,14 +6,20 @@ SHARED_CAFFE_RESOLVER = None def import_caffepb(): - p = os.path.realpath(__file__) + p = os.path.realpath(__file__) 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