ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix amd-smi GPU detection for GPU[N] output format

Browse source 
Older amd-smi versions output "GPU[0] : Card series: ..." instead of
"GPU: 0". The regex now matches both "GPU: <digit>" and "GPU[<digit>"
formats to detect actual GPU data rows.
This commit is contained in:
Daniel Han 2026-04-03 12:51:22 +00:00
commit 86735ffad1
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/install_python_stack.py
View file

@ -112,7 +112,7 @@ def _has_rocm_gpu() -> bool:
# rocminfo: look for "Name: gfxNNNN" indicating an actual GPU agent
(["rocminfo"], lambda out: "gfx" in out.lower()),
# amd-smi list: require "GPU: <number>" data rows, not just a header
(["amd-smi", "list"], lambda out: bool(re.search(r"(?im)^gpu\s*:\s*\d", out))),
(["amd-smi", "list"], lambda out: bool(re.search(r"(?im)^gpu\s*[:\[]\s*\d", out))),
):
exe = shutil.which(cmd[0])
if not exe:
@ -756,7 +756,7 @@ def install_python_stack() -> int:
import re as _re_win
def _win_amd_smi_has_gpu(stdout: str) -> bool:
return bool(_re_win.search(r"(?im)^gpu\s*:\s*\d", stdout))
return bool(_re_win.search(r"(?im)^gpu\s*[:\[]\s*\d", stdout))
_win_amd_gpu = False
for _wcmd, _check_fn in (