diff --git a/gpus.go b/gpus.go index 0173c36..da8098f 100644 --- a/gpus.go +++ b/gpus.go @@ -69,6 +69,7 @@ func ParseTotalGPUs() float64 { descriptor = strings.TrimPrefix(descriptor, "gpu:") descriptor = strings.Split(descriptor, "(")[0] node_gpus, err := strconv.ParseFloat(descriptor, 64) + log.Infof("Number of GPUs %f", node_gpus) if err != nil { num_gpus += node_gpus }