add gpu usage

gpu.go

@@ -16,7 +16,8 @@ func getGPU() ([]DeviceInfo, error) {
 	defer func() {
 		ret := nvml.Shutdown()
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(nvml.ErrorString(ret))
+			fmt.Println("defer", nvml.ErrorString(ret))
+			//return nil, fmt.Errorf(nvml.ErrorString(ret))
 		}
 	}()
 
@@ -104,7 +105,9 @@ func getGpuUsage() ([]DeviceInfo, error) {
 	defer func() {
 		ret := nvml.Shutdown()
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(nvml.ErrorString(ret))
+			fmt.Println("defer", nvml.ErrorString(ret))
+
+			//return nil, fmt.Errorf(nvml.ErrorString(ret))
 			//log.Fatalf("Unable to shutdown NVML: %v", nvml.ErrorString(ret))
 		}
 	}()
@@ -141,7 +144,7 @@ func getGpuUsage() ([]DeviceInfo, error) {
 			Type:  fmt.Sprintf("GPU-%d", i),
 			Model: name,
 			Param: "gpu usage",
-			Power: utilization.Gpu,
+			Power: uint64(utilization.Gpu),
 		}
 
 		res = append(res, e)
@@ -150,7 +153,7 @@ func getGpuUsage() ([]DeviceInfo, error) {
 			Type:  fmt.Sprintf("GPU-%d", i),
 			Model: name,
 			Param: "gpu mem usage",
-			Power: utilization.Memory,
+			Power: uint64(utilization.Memory),
 		}
 
 		res = append(res, e)