diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index ef3e511319..d5f89239e5 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -72,6 +72,7 @@ def _detect_rocm_version() -> tuple[int, int] | None: ) if result.returncode == 0: import re + m = re.search(r"ROCm version:\s*(\d+)\.(\d+)", result.stdout) if m: return int(m.group(1)), int(m.group(2))