Fix parsing of nvcc output

rapidsai · Mar 25, 2024 · d3ca171 · d3ca171
1 parent 5596991
commit d3ca171
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)