From f751306225aa7acf9b0b86bd298ddd4df43f6758 Mon Sep 17 00:00:00 2001 From: Stefan Doerr Date: Fri, 14 Feb 2025 09:46:30 +0200 Subject: [PATCH] no defaults for WITH_CUDA envvar. default is autodetection --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 5c6c6fd7..0b6bdcfa 100644 --- a/setup.py +++ b/setup.py @@ -9,9 +9,9 @@ # If WITH_CUDA is defined -if os.environ.get("WITH_CUDA", "0") == "1": +if os.environ.get("WITH_CUDA") == "1": use_cuda = True -elif os.environ.get("WITH_CUDA", "0") == "0": +elif os.environ.get("WITH_CUDA") == "0": use_cuda = False else: use_cuda = torch.cuda._is_compiled()