--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@
 # This needs to happen before get_extensions() is called
 # Usage: python setup.py install --no-cuda
 #    or: pip install . --no-cuda
-BUILD_NO_CUDA = False
+BUILD_NO_CUDA = os.environ.get("COMFY_KITCHEN_BUILD_NO_CUDA", "0") == "1"
 if "--no-cuda" in sys.argv:
     BUILD_NO_CUDA = True
     sys.argv.remove("--no-cuda")  # Remove so setuptools doesn't complain