From 94f3085afd10d76fa6e0640b5eed1d615b11ecea Mon Sep 17 00:00:00 2001
From: qixiaoning <jony.kee@outlook.com>
Date: 星期五, 25 七月 2025 18:44:38 +0800
Subject: [PATCH] 修复标题
---
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