(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211190694.0 (22)申请日 2022.09.28 (71)申请人 苏州浪潮智能科技有限公司 地址 215100 江苏省苏州市吴中经济开发 区郭巷街道官浦路1号9幢 (72)发明人 邢良占　 (74)专利代理 机构 北京集佳知识产权代理有限 公司 11227 专利代理师 崔清杨 (51)Int.Cl. G06F 9/445(2018.01) G06N 20/00(2019.01) (54)发明名称 一种训练任务环境变量设置方法、 装置、 设 备及介质 (57)摘要 本申请公开了一种训练任务环境变量设置 方法、 装置、 设备及介质， 涉及计算机技术领域， 包括： 接收用户创建训练任务时设置的环境变量 数据； 从环 境变量数据中获取与训练任务相关的 全局环境变量， 并利用环境变量数据对全局环境 变量进行处理， 得到处理后环境变量； 将处理后 环境变量注入到任务容器中， 以便在执行训练任 务时， 通过训练任务中的训练脚本直接从所述任 务容器中调用所述处理后环境变量。 本申请在创 建训练任务时设置环境变量， 并将设置的环境变 量注入到任务容器中， 这样便不需要在训练脚本 中设置环 境变量， 在训练任务运行时直接从任务 容器中获取就可以了， 进而提升了训练任务的训 练效率， 解决了训练任务使用环 境变量时的易用 性问题。 权利要求书2页 说明书9页 附图4页 CN 115509633 A 2022.12.23 CN 115509633 A 1.一种训练任务环境变量设置方法， 其特 征在于， 应用于深度学习服 务平台， 包括： 接收用户创建训练任务时设置的环境变量数据； 从所述环境变量数据中获取与所述训练任务相关的全局 环境变量， 并利用所述环境变 量数据对所述全局环境变量进行处 理， 得到处 理后环境变量； 将所述处理后环境变量注入到任务容器中， 以便在执行所述训练任务时， 通过所述训 练任务中的训练脚本直接从所述任务 容器中调用所述处 理后环境变量。 2.根据权利要求1所述的训练任务环境变量设置方法， 其特征在于， 所述接收用户创建 训练任务时设置的环境变量数据， 包括： 接收用户创建训练任务时通过页面设置的方式设置的环境变量数据。 3.根据权利要求1所述的训练任务环境变量设置方法， 其特 征在于， 还 包括： 在所述深度学习服务平台的系统设置中设置所述全局环境变量； 其中， 所述全局环境 变量包括所述训练任务默认注入的环境变量构成的第一环境变量列表和所述训练任务不 能设置的第二环境变量列表。 4.根据权利要求2所述的训练任务环境变量设置方法， 其特征在于， 所述从所述环境变 量数据中获取与所述训练任务相关的全局环境变量， 并利用所述环境变量数据对所述全局 环境变量进行处 理， 得到处 理后环境变量， 包括： 从所述环境变量数据中获取所述训练任务默认注入的环境变量构成的第一环境变量 列表和所述训练任务 不能设置的第二环境变量列表； 利用所述环境变量数据对所述第 一环境变量列表和所述第 二环境变量列表进行设置， 得到处理后环境变量。 5.根据权利要求1所述的训练任务环境变量设置方法， 其特征在于， 所述接收用户创建 训练任务时设置的环境变量数据， 包括： 通过预先创建的任务环境变量处理模块接收用户创建训练任务时设置的环境变量数 据。 6.根据权利要求5所述的训练任务环境变量设置方法， 其特征在于， 所述从所述环境变 量数据中获取与所述训练任务相关的全局环境变量， 包括： 通过所述任务环境变量处理模块对所述环境变量数据进行校验， 若校验通过则从所述 环境变量数据中获取与所述训练任务相关的全局环境变量。 7.根据权利要求1至 6任一项所述的训练任务环境变量设置方法， 其特 征在于， 还 包括： 通过页面设置的方式对所述任务容器中的所述处理后环境变量进行增加、 删除和修 改。 8.一种训练任务环境变量设置装置， 应用于深度学习服 务平台， 其特 征在于， 包括： 环境变量接收模块， 用于 接收用户创建训练任务时设置的环境变量数据； 全局环境变量获取模块， 用于从所述环境变量数据中获取与 所述训练任务相关的全局 环境变量； 环境变量处理模块， 用于利用所述环境变量数据对所述全局环境变量进行处理， 得到 处理后环境变量； 环境变量注入 模块， 用于将所述处 理后环境变量注入到任务 容器中； 环境变量调用模块， 用于在执行所述训练任务时， 通过所述训练任务中的训练脚本直权　利　要　求　书 1/2 页 2 CN 115509633 A 2接从所述任务 容器中调用所述处 理后环境变量。 9.一种电子设备， 其特征在于， 包括处理器和存储器； 其中， 所述处理器执行所述存储 器中保存的计算机程序时实现如权利要求1至7任一项所述的训练任务环境变量设置方法。 10.一种计算机可读存储介质， 其特征在于， 用于存储计算机程序； 其中， 所述计算机程 序被处理器执行时实现如权利要求1至7任一项所述的训练任务环境变量设置方法。权　利　要　求　书 2/2 页 3 CN 115509633 A 3