diff --git a/setup.py b/setup.py index 1284f9f5d7..ba62ea2671 100644 --- a/setup.py +++ b/setup.py @@ -161,6 +161,9 @@ def get_extensions(): define_macros += [('MMCV_WITH_TRT', None)] cuda_args = os.getenv('MMCV_CUDA_ARGS') extra_compile_args['nvcc'] = [cuda_args] if cuda_args else [] + # prevent cub/thrust conflict with other python library + # More context See issues #1454 + extra_compile_args['nvcc'] += ['-Xcompiler=-fno-gnu-unique'] library_dirs += library_paths(cuda=True) from setuptools import Extension