log gpu power

d9302e92 · Ubuntu · c2741254 · d9302e92
Commit d9302e92 authored Mar 20, 2024 by Ubuntu
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

gpu.go gpu.go +4 -0

No files found.
--- a/gpu.go
+++ b/gpu.go
@@ -7,6 +7,8 @@ import (
 	"github.com/NVIDIA/go-nvml/pkg/nvml"
 )

+// NVIDIA GeForce RTX 3080
+
 func getGPU() ([]DeviceInfo, error) {
 	ret := nvml.Init()
 	if ret != nvml.SUCCESS {
@@ -68,6 +70,8 @@ func getGPU() ([]DeviceInfo, error) {

 		if v, ok := comput[name]; ok {
 			c = v
+		} else {
+			fmt.Println("can not find the model", name)
 		}

 		e = DeviceInfo{