From d3ca1715ae6636432862c16242a320066e996312 Mon Sep 17 00:00:00 2001 From: Kyle Edwards Date: Mon, 25 Mar 2024 17:09:37 -0400 Subject: [PATCH] Fix parsing of nvcc output --- rapids_build_backend/impls.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rapids_build_backend/impls.py b/rapids_build_backend/impls.py index c4fe037..de18847 100644 --- a/rapids_build_backend/impls.py +++ b/rapids_build_backend/impls.py @@ -63,7 +63,7 @@ def _get_cuda_version(require_cuda=False): output_lines = process_output.stdout.decode().splitlines() - match = re.search(r"release (\d+\.\d+)", output_lines[3]) + match = re.search(r"release (\d+)\.(\d+)", output_lines[3]) if match is None: raise ValueError("Failed to parse CUDA version from nvcc output.") return match.group(1), match.group(2)