Correct location of warning

dirac-institute · Sep 14, 2023 · 1f8e586 · 1f8e586
1 parent 7f2a94e
commit 1f8e586
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -132,9 +132,9 @@ def build_extension(self, ext: CMakeExtension) -> None:
         try:
             subprocess.check_output('nvidia-smi')
             cmake_args += ["-DCPU_ONLY=OFF"]
-            print("WARNING: No GPU Found. Building with CPU only mode.")
         except Exception:
             cmake_args += ["-DCPU_ONLY=ON"]
+            print("WARNING: No GPU Found. Building with CPU only mode.")
 
         # Set CMAKE_BUILD_PARALLEL_LEVEL to control the parallel build level
         # across all generators.