Only consider display class for gpus

metal-stack · Oct 15, 2024 · b0371a4 · b0371a4
1 parent b095b39
commit b0371a4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/register/register.go b/cmd/register/register.go
@@ -245,8 +245,11 @@ func (r *Register) detectGPUs() (pci.Devices, error) {
 		if !strings.Contains(strings.ToLower(device.VendorName), "nvidia") {
 			continue
 		}
+		r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName, "classname", device.ClassName)
 
-		r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName)
+		if strings.ToLower(device.ClassName) != "display" {
+			continue
+		}
 		result = append(result, device)
 	}