enable gpu monitoring by default

This commit is contained in:
Henry Dollman
2024-11-12 18:13:57 -05:00
parent 2e8ac98924
commit b1db450e00
2 changed files with 7 additions and 9 deletions

View File

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