diff --git a/setup.py b/setup.py
index 3df847dd5b..ced20fd156 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,16 @@
         from torch_mlu.utils.cpp_extension import BuildExtension
         EXT_TYPE = 'pytorch'
     else:
-        from torch.utils.cpp_extension import BuildExtension
+        # build disable ninja
+        if os.getenv("MMCV_NO_NINJA", "0") == "1":
+        
+            class BuildExtension(torch.utils.cpp_extension.BuildExtension):
+                def __init__(self, *args, **kwargs):
+                    super().__init__(use_ninja=False, *args, **kwargs)
+        
+        else:
+            BuildExtension = torch.utils.cpp_extension.BuildExtension
+
         EXT_TYPE = 'pytorch'
     cmd_class = {'build_ext': BuildExtension}
 except ModuleNotFoundError: