gpu usage and vram charts

beszel/internal/agent/agent.go

@@ -76,10 +76,12 @@ func (a *Agent) Run(pubKey []byte, addr string) {
 	a.dockerManager = newDockerManager()
 
 	// initialize GPU manager
-	if gm, err := NewGPUManager(); err != nil {
-		slog.Error("GPU manager", "err", err)
-	} else {
-		a.gpuManager = gm
+	if os.Getenv("GPU") == "true" {
+		if gm, err := NewGPUManager(); err != nil {
+			slog.Error("GPU manager", "err", err)
+		} else {
+			a.gpuManager = gm
+		}
 	}
 
 	// if debugging, print stats

beszel/internal/agent/gpu.go

@@ -94,8 +94,8 @@ func (gm *GPUManager) parseNvidiaData(output []byte) {
 				// update gpu data
 				gpu := gm.GpuDataMap[id]
 				gpu.Temperature += temp
-				gpu.MemoryUsed += memoryUsage
-				gpu.MemoryTotal += totalMemory
+				gpu.MemoryUsed += memoryUsage / 1.024
+				gpu.MemoryTotal += totalMemory / 1.024
 				gpu.Usage += usage
 				gpu.Power += power
 				gpu.Count++