diff --git a/setup.py b/setup.py
index d1f125e6b4f34502bb708fb6db17c7900351b9c5..daa9dbd5cc6e40297ad862adaf06847ff1902bf4 100644
--- a/setup.py
+++ b/setup.py
@@ -685,6 +685,11 @@ def options_process(args, build_options):
 def get_cmake_generator():
     if os.getenv("GENERATOR"):
         cmake_generator = os.getenv("GENERATOR")
+        if os.system('ninja --version') == 0:
+            print("Ninja has been installed,use ninja to compile Paddle now.")
+        else:
+            print("Ninja has not been installed,install it now.")
+            os.system('python -m pip install ninja')
     else:
         cmake_generator = "Unix Makefiles"
     return cmake_generator