From 5e0d2a367230435bc973b33d951e05920296cf14 Mon Sep 17 00:00:00 2001 From: kuizhiqing Date: Tue, 16 Nov 2021 16:07:44 +0800 Subject: [PATCH] replace VISIBLE_IDS by ASCEND_VISIBLE_DEVICES --- cmd/deviceplugin/server.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/deviceplugin/server.go b/cmd/deviceplugin/server.go index ccec634..595828b 100644 --- a/cmd/deviceplugin/server.go +++ b/cmd/deviceplugin/server.go @@ -182,7 +182,7 @@ func (srv *server) Allocate(ctx context.Context, rqt *pluginapi.AllocateRequest) cresp.Envs = make(map[string]string) } - cresp.Envs["VISIBLE_IDS"] = strings.Join(npuList, ",") + cresp.Envs["ASCEND_VISIBLE_DEVICES"] = strings.Join(npuList, ",") if ips, err := srv.getDeviceIP(npuList); err == nil { cresp.Envs["NPU_IPS"] = ips -- Gitee