re-enable nvml

beam-cloud · Jan 19, 2024 · fc6d3d3 · fc6d3d3
1 parent 66474cd
commit fc6d3d3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/internal/worker/metrics_linux.go b/internal/worker/metrics_linux.go
@@ -9,9 +9,7 @@ import (
 )
 
 func (wm *WorkerMetrics) InitNvml() {
-	// TODO: investigate segmentation violation
-	// wm.nvmlActive = nvml.Init() == nvml.SUCCESS
-	wm.nvmlActive = false
+	wm.nvmlActive = nvml.Init() == nvml.SUCCESS
 }
 
 func (wm *WorkerMetrics) Shutdown() {