in pkg/gpu/nvidia/manager.go [286:302]
func (ngm *nvidiaGPUManager) discoverNumGPUs() (int, error) {
reg := regexp.MustCompile(nvidiaDeviceRE)
deviceCount := 0
files, err := ioutil.ReadDir(ngm.devDirectory)
if err != nil {
return deviceCount, err
}
for _, f := range files {
if f.IsDir() {
continue
}
if reg.MatchString(f.Name()) {
deviceCount++
}
}
return deviceCount, nil
}