feat: 添加 GPU 名称获取功能

This commit is contained in:
Akizon77
2025-05-26 19:12:02 +08:00
parent 6257278426
commit 21b1b17e84
4 changed files with 100 additions and 1 deletions

View File

@@ -0,0 +1,32 @@
//go:build linux
// +build linux
package monitoring
import (
"os/exec"
"strings"
)
func GpuName() string {
accept := []string{"vga", "nvidia", "amd", "radeon", "render"}
out, err := exec.Command("lspci").Output()
if err == nil {
lines := strings.Split(string(out), "\n")
for _, line := range lines {
for _, a := range accept {
if strings.Contains(strings.ToLower(line), a) {
parts := strings.SplitN(line, ":", 4)
if len(parts) >= 4 {
return strings.TrimSpace(parts[3])
} else if len(parts) == 3 {
return strings.TrimSpace(parts[2])
} else if len(parts) == 2 {
return strings.TrimSpace(parts[1])
}
}
}
}
}
return "None"
}