From a5d0373df88f6be8c6a5eaafd3f22b5bca1d88a6 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Mon, 3 Feb 2025 11:37:55 +0100 Subject: [PATCH] Fix build without CUDA --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 162db7593..d00bff197 100644 --- a/setup.py +++ b/setup.py @@ -134,7 +134,8 @@ def check_cudnn_version_and_warn(global_option: str, required_cudnn_version: int version_ge_1_5 = ["-DVERSION_GE_1_5"] version_dependent_macros = version_ge_1_1 + version_ge_1_3 + version_ge_1_5 -_, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME) +if CUDA_HOME is not None: + _, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME) if "--distributed_adam" in sys.argv: sys.argv.remove("--distributed_adam")