From fe724b50b3f1b3dfe2219eb9af4bcca96c89a158 Mon Sep 17 00:00:00 2001
From: qixiaoning <jony.kee@outlook.com>
Date: 星期二, 08 七月 2025 18:19:22 +0800
Subject: [PATCH] 2025/7/8 修改获取算力为登临
---
extend/util/util.go | 13 +++++++------
1 files changed, 7 insertions(+), 6 deletions(-)
diff --git a/extend/util/util.go b/extend/util/util.go
index 60c8860..c467ac3 100644
--- a/extend/util/util.go
+++ b/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("鏃犲彲鐢℅PU!璇锋鏌ユ槸鍚︽甯稿畨鏄惧崱鍙婃樉鍗￠┍鍔紝閫氳繃nvidia-smi鏌ヨ鐘舵��")
+ return vGpu, errors.New("鏃犲彲鐢℅PU!璇锋鏌ユ槸鍚︽甯稿畨鏄惧崱鍙婃樉鍗￠┍鍔紝閫氳繃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("鏃犲彲鐢℅PU!璇锋鏌ユ槸鍚︽甯稿畨鏄惧崱鍙婃樉鍗￠┍鍔紝閫氳繃nvidia-smi鏌ヨ鐘舵��")
+ return vGpu, errors.New("鏃犲彲鐢℅PU!璇锋鏌ユ槸鍚︽甯稿畨鏄惧崱鍙婃樉鍗￠┍鍔紝閫氳繃dlsmi鏌ヨ鐘舵��")
}
return vGpu, nil
}
--
Gitblit v1.8.0