add gpu number

This commit is contained in:
Link 2024-03-01 09:09:05 +00:00
parent 62006f61b5
commit 7bd0df6441

View File

@ -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,