diff --git a/cmd/deviceplugin/server.go b/cmd/deviceplugin/server.go index ccec634adcf712a56c0c3c69be0ecb83e67e604d..595828bdd6c9f8aa069fec8b9123f2e4e81055d4 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