optional import pytorch rewriter

Author: grimoire

mmdeploy/__init__.py

@@ -4,7 +4,11 @@
 from mmdeploy.utils import get_root_logger
 from .version import __version__  # noqa F401
 
-importlib.import_module('mmdeploy.pytorch')
+if importlib.util.find_spec('torch'):
+    importlib.import_module('mmdeploy.pytorch')
+else:
+    logger = get_root_logger()
+    logger.debug('torch is not installed.')
 
 if importlib.util.find_spec('mmcv'):
     importlib.import_module('mmdeploy.mmcv')

mmdeploy/backend/sdk/__init__.py

@@ -20,10 +20,13 @@
     lib_dir = os.path.dirname(lib_path)
     sys.path.insert(0, lib_dir)
 
-if importlib.util.find_spec(module_name) is not None:
-    from .wrapper import SDKWrapper
-    __all__ = ['SDKWrapper']
-    _is_available = True
+try:
+    if importlib.util.find_spec(module_name) is not None:
+        from .wrapper import SDKWrapper
+        __all__ = ['SDKWrapper']
+        _is_available = True
+except Exception:
+    pass
 
 
 def is_available() -> bool:

requirements/runtime.txt

@@ -1,5 +1,6 @@
 h5py
 matplotlib
+multiprocess
 numpy
 onnx>=1.8.0
 six

setup.py

@@ -1,16 +1,7 @@
-import logging
-
 from setuptools import find_packages, setup
 
 version_file = 'mmdeploy/version.py'
 
-try:
-    from torch.utils.cpp_extension import BuildExtension
-    cmd_class = {'build_ext': BuildExtension}
-except ModuleNotFoundError:
-    cmd_class = {}
-    logging.warning('Skip building ext ops due to the absence of torch.')
-
 
 def readme():
     with open('README.md', encoding='utf-8') as f:
@@ -135,5 +126,5 @@ def gen_packages_items():
             'optional': parse_requirements('requirements/optional.txt'),
         },
         ext_modules=[],
-        cmdclass=cmd_class,
+        cmdclass={},
         zip_safe=False)