qixiaoning
2025-07-08 fe724b50b3f1b3dfe2219eb9af4bcca96c89a158
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
}