hoangvv/NextZen-UserService
1
0
Fork 0
mirror of https://github.com/KaySar12/NextZen-UserService.git synced 2025-10-06 19:59:42 +07:00
Code Issues Packages Projects Releases Wiki Activity

add gpu number

Browse Source
This commit is contained in:
Link
2024-03-01 09:09:05 +00:00
parent 62006f61b5
commit 7bd0df6441
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
route/v1/user.go
+6
View File
@@ -19,6 +19,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/IceWhaleTech/CasaOS-Common/external"
"github.com/IceWhaleTech/CasaOS-Common/utils/common_err" "github.com/IceWhaleTech/CasaOS-Common/utils/common_err"
"github.com/IceWhaleTech/CasaOS-Common/utils/jwt" "github.com/IceWhaleTech/CasaOS-Common/utils/jwt"
"github.com/IceWhaleTech/CasaOS-Common/utils/logger" "github.com/IceWhaleTech/CasaOS-Common/utils/logger"
@@ -818,6 +819,11 @@ func GetUserStatus(c *gin.Context) {
data["key"] = key data["key"] = key
data["initialized"] = false 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, c.JSON(common_err.SUCCESS,
model.Result{ model.Result{
Success: common_err.SUCCESS, Success: common_err.SUCCESS,