From 1dee4eae03db6c6ab028b2a37af07a9c6f0bb226 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 23 十二月 2019 16:21:35 +0800
Subject: [PATCH] update;
---
app/slave/slave.go | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/app/slave/slave.go b/app/slave/slave.go
index 6c77ec2..bb084fc 100644
--- a/app/slave/slave.go
+++ b/app/slave/slave.go
@@ -6,6 +6,8 @@
"analysis/util"
"context"
"plugin"
+
+ "basic.com/valib/gogpu.git"
)
// TwoPluginConflict test
@@ -39,6 +41,14 @@
return false
}
+ // 閰嶇疆鏂囦欢璁剧疆浜唃pu memory涓斿�艰嚦灏戜袱浣嶆暟,鍒ゆ柇鏄惁鏈塯pu鍙敤
+ if v, ok := cfg.Param["gpu-memory"]; ok && len(v) > 1 {
+ indices := gogpu.RankGPU()
+ if len(indices) == 0 {
+ logo.Errorln("!!!!!!THERE IS NO VALID GPU")
+ return false
+ }
+ }
// func Create(config string, typ, id string, gpu int, shm bool, ipc2Rule string, ruleMaxSize int, fn func(...interface{}), reserved map[string]string) interface{}
// func Run(ctx context.Context, i interface{}) {
--
Gitblit v1.8.0