Skip to content

Commit 2faca7a

Browse files
committed
parse the return value for the gpu name
1 parent c039c96 commit 2faca7a

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

src/utils/core/gpu_install.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,11 @@ def fetch_gpu(log: Logger) -> tuple[str, str | None] | None:
3131
log.logger("e", f"{Err}. Command lspci failed to execute.")
3232
return None
3333
else:
34-
return gpu_name.decode("utf-8").strip().split("\n")
34+
return [
35+
gpu.split(":") for gpu in (
36+
gpu_name
37+
.decode("utf-8")
38+
.strip()
39+
.split("\n")
40+
)
41+
]

0 commit comments

Comments
 (0)