gpu usage and vram charts

This commit is contained in:
Henry Dollman
2024-11-08 18:00:30 -05:00
parent 8262a9a45b
commit cd10727795
7 changed files with 82 additions and 16 deletions

View File

@@ -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