| | |
| | | package util |
| | | |
| | | import ( |
| | | "basic.com/valib/logger.git" |
| | | "bytes" |
| | | "crypto/aes" |
| | | "crypto/cipher" |
| | |
| | | "encoding/hex" |
| | | "encoding/json" |
| | | "fmt" |
| | | "github.com/gin-gonic/gin" |
| | | "github.com/pkg/errors" |
| | | "io" |
| | | "io/ioutil" |
| | | "net" |
| | |
| | | "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" |
| | |
| | | 获取英伟达显卡的型号 |
| | | */ |
| | | 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 |
| | | } |