fix tensorrt test failed with pytorch 1.8+ (#1464)

* fix tensorrt test failed with pytorch 1.8+ * add comment
open-mmlab · Dec 10, 2021 · 48f44b6 · 48f44b6
1 parent 222f380
commit 48f44b6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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