2025/7/8 修改获取算力为登临

qixiaoning

2025-07-08 fe724b50b3f1b3dfe2219eb9af4bcca96c89a158

2025/7/8 修改获取算力为登临

 extend/util/util.go

@@ -1,7 +1,6 @@
package util

import (
    "basic.com/valib/logger.git"
    "bytes"
    "crypto/aes"
    "crypto/cipher"
@@ -12,8 +11,6 @@
    "encoding/hex"
    "encoding/json"
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/pkg/errors"
    "io"
    "io/ioutil"
    "net"
@@ -29,6 +26,10 @@
    "time"
    "vamicro/config"
    "vamicro/extend/code"

    "basic.com/valib/logger.git"
    "github.com/gin-gonic/gin"
    "github.com/pkg/errors"
)

const TimeFmtStr = "2006-01-02 15:04:05"
@@ -433,15 +434,15 @@
获取英伟达显卡的型号
*/
func NvidiaVGpu() (vGpu string, err error) {
    resultMsg, e := ExeCmd("nvidia-smi -L")
    resultMsg, e := ExeCmd("dlsmi -l 1")
    if e != nil {
        return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过nvidia-smi查询状态")
        return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过dlsmi查询状态")
    }
    indeces := strings.Split(string(resultMsg), "\n")[0]
    if len(indeces) > 0 {
        vGpu = strings.Split(strings.Split(indeces, " (")[0], ": ")[1]
    } else {
        return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过nvidia-smi查询状态")
        return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过dlsmi查询状态")
    }
    return vGpu, nil
}