diff --git a/pkg/hostman/isolated_device/gpu.go b/pkg/hostman/isolated_device/gpu.go index e41d21670d..27c86cb6cc 100644 --- a/pkg/hostman/isolated_device/gpu.go +++ b/pkg/hostman/isolated_device/gpu.go @@ -95,7 +95,7 @@ func getPassthroughGPUs(filteredAddrs []string, enableWhitelist bool, whitelistM if !utils.IsInArray(dev.ClassCode, GpuClassCodes) { continue } - if !utils.IsInStringArray(dev.VendorId, []string{api.NVIDIA_VENDOR_ID, api.AMD_VENDOR_ID}) { + if dev.ClassCode == CLASS_CODE_DISP && !utils.IsInStringArray(dev.VendorId, []string{api.NVIDIA_VENDOR_ID, api.AMD_VENDOR_ID}) { log.Infof("Skip add device %s vendor is unsupport", dev.Addr) continue }