fix gpu

gpu.go

@@ -24,7 +24,7 @@ func getGPU() ([]DeviceInfo, error) {
 		log.Fatalf("Unable to get device count: %v", nvml.ErrorString(ret))
 	}
 
-	res := make([]DeviceInfo, 0, len(count))
+	res := make([]DeviceInfo, 0, count)
 
 	for i := 0; i < count; i++ {
 
@@ -33,10 +33,10 @@ func getGPU() ([]DeviceInfo, error) {
 			log.Fatalf("Unable to get device at index %d: %v", i, nvml.ErrorString(ret))
 		}
 
-		memory, err := device.GetMemoryInfo()
-		if err != nvml.SUCCESS {
-			panic(err)
-		}
+		// memory, err := device.GetMemoryInfo()
+		// if err != nvml.SUCCESS {
+		// 	panic(err)
+		// }
 
 		// d1, _, err := device.GetDriverModel()
 		// if err != nvml.SUCCESS {
@@ -48,13 +48,18 @@ func getGPU() ([]DeviceInfo, error) {
 			panic(err)
 		}
 
+		memory, err := device.GetMemoryInfo()
+		if err != nvml.SUCCESS {
+			fmt.Println(err)
+		}
+
 		//fmt.Printf(" %v\n", proc)
 
 		e := DeviceInfo{
 			Type:  fmt.Sprintf("GPU-%d", i),
 			Model: name,
 			Param: "device.GetMemoryInfo()",
-			Power: uint64(device.GetMemoryInfo()),
+			Power: uint64(memory.Total),
 		}
 
 		res = append(res, e)