diff --git a/third_party/gpus/find_cuda_config.py b/third_party/gpus/find_cuda_config.py index faabd827def..90265f60773 100644 --- a/third_party/gpus/find_cuda_config.py +++ b/third_party/gpus/find_cuda_config.py @@ -116,7 +116,7 @@ def _at_least_version(actual_version, required_version): def _get_header_version(path, name): """Returns preprocessor defines in C header file.""" for line in open(path, "r").readlines(): - match = re.match("#define %s (\d+)" % name, line) + match = re.match("#define %s +(\d+)" % name, line) if match: return match.group(1) return ""