From 1f8e58650fb1c25c57cf2ab585bd4c89e895589b Mon Sep 17 00:00:00 2001
From: Jeremy Kubica <104161096+jeremykubica@users.noreply.github.com>
Date: Thu, 14 Sep 2023 15:16:39 -0400
Subject: [PATCH] Correct location of warning

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index a9b3fdddd..78caa1a41 100644
--- 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.