add gpu number

IceWhaleTech · Mar 1, 2024 · 7bd0df6 · 7bd0df6
1 parent 62006f6
commit 7bd0df6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/route/v1/user.go b/route/v1/user.go
@@ -19,6 +19,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/IceWhaleTech/CasaOS-Common/external"
 	"github.com/IceWhaleTech/CasaOS-Common/utils/common_err"
 	"github.com/IceWhaleTech/CasaOS-Common/utils/jwt"
 	"github.com/IceWhaleTech/CasaOS-Common/utils/logger"
@@ -818,6 +819,11 @@ func GetUserStatus(c *gin.Context) {
 		data["key"] = key
 		data["initialized"] = false
 	}
+	gpus, err := external.NvidiaGPUInfoList()
+	if err != nil {
+		logger.Error("NvidiaGPUInfoList error", zap.Error(err))
+	}
+	data["gpus"] = len(gpus)
 	c.JSON(common_err.SUCCESS,
 		model.Result{
 			Success: common_err.SUCCESS,