check protoc version

tensorpack/utils/concurrency.py

@@ -193,21 +193,27 @@ def start_proc_mask_signal(proc):
 def subproc_call(cmd, timeout=None):
     """
     Execute a command with timeout, and return both STDOUT/STDERR.
+
     Args:
         cmd(str): the command to execute.
         timeout(float): timeout in seconds.
+
+    Returns:
+        output(str), retcode(int). If timeout, retcode is -1.
     """
     try:
         output = subprocess.check_output(
             cmd, stderr=subprocess.STDOUT,
             shell=True, timeout=timeout)
-        return output
+        return output, 0
     except subprocess.TimeoutExpired as e:
         logger.warn("Command timeout!")
         logger.warn(e.output)
+        return e.output, -1
     except subprocess.CalledProcessError as e:
         logger.warn("Commnad failed: {}".format(e.returncode))
         logger.warn(e.output)
+        return e.output, e.returncode
 
 
 class OrderedContainer(object):

tensorpack/utils/loadcaffe.py

@@ -3,11 +3,13 @@
 # File: loadcaffe.py
 # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
 
+import sys
 import numpy as np
 import os
 
 from .utils import change_env
 from .fs import download, get_dataset_path
+from .concurrency import subproc_call
 from . import logger
 
 __all__ = ['load_caffe', 'get_caffe_pb']
@@ -123,6 +125,20 @@ def get_caffe_pb():
     if not os.path.isfile(caffe_pb_file):
         download(CAFFE_PROTO_URL, dir)
         assert os.path.isfile(os.path.join(dir, 'caffe.proto'))
+
+        if sys.version_info.major == 3:
+            cmd = "protoc --version"
+            version, ret = subproc_call(cmd, timeout=3)
+            if ret != 0:
+                sys.exit(1)
+            try:
+                version = version.decode('utf-8')
+                version = float('.'.join(version.split(' ')[1].split('.')[:2]))
+                assert version >= 2.7, "Require protoc>=2.7 for Python3"
+            except:
+                logger.exception("protoc --version gives: " + str(version))
+                raise
+
         cmd = 'cd {} && protoc caffe.proto --python_out .'.format(dir)
         ret = os.system(cmd)
         assert ret == 0, \