From 0a324c77e33d8c0b61fd6d166cc8d3cec7d78139 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 10 七月 2019 18:29:23 +0800
Subject: [PATCH] update
---
gogpu.go | 27 +++++++++++++++++++++++++--
1 files changed, 25 insertions(+), 2 deletions(-)
diff --git a/gogpu.go b/gogpu.go
index 25f7c71..86a7605 100644
--- a/gogpu.go
+++ b/gogpu.go
@@ -11,16 +11,39 @@
import (
"encoding/json"
"errors"
+ "fmt"
"strconv"
"strings"
"unsafe"
)
-// IdleGPU pass needed gpu memory size test if satisfy
-func IdleGPU(memSize int) int {
+// ValidGPU pass needed gpu memory size test if satisfy
+func ValidGPU(memSize int) int {
return int(C.get_idle_gpu(C.int(memSize)))
}
+// IdleGPU idle most
+func IdleGPU() int {
+ info, err := Info()
+ if err != nil {
+ fmt.Println("no gpu")
+ return -1
+ }
+ if info.Count == 1 {
+ return 0
+ }
+
+ var free int64
+ gpu := 0
+ for k, v := range info.Info {
+ if v.GpuMemoryFree > free {
+ free = v.GpuMemoryFree
+ gpu = k
+ }
+ }
+ return gpu
+}
+
// GpuUnitInfo gpu card info
type GpuUnitInfo struct {
GpuUtilization int `json:"utilization"`
--
Gitblit v1.8.0