网易首页 > 网易号 > 正文 申请入驻

一文详述华为最新开源框MindSpore分布式自动并行训练

0
分享至

4 月 23 日,机器之心联合华为昇腾学院开设的线上公开课《轻松上手开源框架 MindSpore》第一课正式开讲,MindSpore 前端开发工程师龚子妍为大家带来了主题分享《MindSpore 分布式自动并行训练》。

在第一课《MindSpore分布式自动并行训练》中,我们主要介绍了几种相关的并行模式,包括数据并行,通过分发数据集的方式来提升训练效率;模型并行,通过拆分模型参数来突破单卡的内存瓶颈;混合并行是融合了数据并行和模型并行的一种并行模式。

这以上的并行模式在 MindSpore中 都是依赖集合通信来完成多卡间数据同步的。而自动并行特性主要的特点有:通过算子自动切分,推导张量的排布模型;利用构建代价模型的方式,自动搜索了切分策略;并将自动微分扩展到了分布式的领域,可以自动求导通信算子的反向梯度。

MindSpore的分布式训练,支持数字并行、模型并行、自动并行的多层次接口,其中,自动并行更加易用,可以有效的帮助网络提升性能。

在第一课的 QA 环节中,有一些问题被大家广泛提到,我们再次做了精选与编辑,供大家参考。

Q1:切分是平均切分吗? 如果是 5 个分四份会怎样?

首先就是在切分方式这里,其实是有并行切分的约束的,比如说它的维度的数据和计算资源其实是需要均匀切分,是以 2 为基,也就是说指定的切分策略中的数字必须都是 2 的倍数。另外像刚刚提到的,如果无法均分的问题,其实在框架中我们是会做校验的,如果是出现了没有办法均分的张量排布模型,我们会在校验中直接报错。

Q2:需要什么硬件支持呢?自己的笔记本电脑是否可以来执行网络?

目前来说的话笔记本电脑或者有 GPU 的环境都可以通过 docker 镜像来简单试用,至于自动并行的特性,对 CPU 和 GPU 的支持还在完善中。我们推荐用户在昇腾芯片上使用自动并行,作为普通用户的话,可以关注我们的开源社区,可以去申请 MindSpore 开发者体验环境进行试用。

Q3:效率具体上有多大提升?有对照数据吗

在开源社区我们公布了 Cifar10 数据集的训练性能数据。对于自动并行特性的话,和业界横向对比的数据还在测试收集中,目前公布的性能数据,是基于 mindspore+昇腾芯片,和其他的并行模式做的比较。从刚刚展示的表格中其实可以看到,自动并行在大分类数的情况下相比数据并行模式性能有比较大的提升。

Q4:讲一下 pipeline 并行?

在我们的框架规划图上是可以看到是有提到 pipeline 并行这个概念的,是自动并行特性其中的一个规划方向。pipeline 并行,也就是我们刚刚提到的层间模型并行的概念,需要把模型按层切分到多个卡上,如果让用户手动实现对开发者来说难度非常大,需要考虑性能、内存、通信、训练效果等问题,我们期望通过框架完成模型按层自动切分到卡上的目标。

Q5:请问数据并行是 allreduce 实现吗?

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
他是强渡乌江的功臣,不满职务离开八路军,毛主席说情:留条活路

他是强渡乌江的功臣,不满职务离开八路军,毛主席说情:留条活路

大海的诺言
2024-10-08 15:04:01
韩女星朴娜莱和男性朋友在车上发生了性关系,两名经纪人被迫围观

韩女星朴娜莱和男性朋友在车上发生了性关系,两名经纪人被迫围观

西楼知趣杂谈
2026-04-09 09:35:17
现在还敢辞掉工作回农村老家的,90%是这4类人,你属于哪一种呢?

现在还敢辞掉工作回农村老家的,90%是这4类人,你属于哪一种呢?

三农老历
2026-04-28 18:27:04
河北一中学教师买卖8支玩具仿真枪获刑三年,申诉被驳回

河北一中学教师买卖8支玩具仿真枪获刑三年,申诉被驳回

澎湃新闻
2026-04-28 16:06:27
为何超市月薪3000元,却没人肯辞职?超市员工:“傻子”才辞职

为何超市月薪3000元,却没人肯辞职?超市员工:“傻子”才辞职

平说财经
2026-04-26 23:06:17
白宫晚宴枪手曝光:发言极端激进!仇恨基督教,系无组织独狼袭击

白宫晚宴枪手曝光:发言极端激进!仇恨基督教,系无组织独狼袭击

小小科普员
2026-04-27 14:33:37
黑猴也被干趴了!《黑神话》D加密终被破解:历时一年半

黑猴也被干趴了!《黑神话》D加密终被破解:历时一年半

游民星空
2026-04-28 11:06:20
深夜重磅!今天,伊朗赢得重大喜讯

深夜重磅!今天,伊朗赢得重大喜讯

阿振观点
2026-04-28 00:17:31
8轮过后,中超5名主帅帅位不稳,3洋帅2土帅,谁会第一个下课?

8轮过后,中超5名主帅帅位不稳,3洋帅2土帅,谁会第一个下课?

我就是一个说球的
2026-04-27 22:03:53
笑死在张若昀的评论区,网友:一个没参加浪姐却无处不在的人

笑死在张若昀的评论区,网友:一个没参加浪姐却无处不在的人

一盅情怀
2026-04-27 13:45:59
医生发现:老人若长时间不吃甜食,用不了多长时间身体有5大改善

医生发现:老人若长时间不吃甜食,用不了多长时间身体有5大改善

芹姐说生活
2026-04-23 15:04:44
马龙送国乒出征世乒赛,温瑞博陈熠随行!国际乒联发布王楚钦海报

马龙送国乒出征世乒赛,温瑞博陈熠随行!国际乒联发布王楚钦海报

小彭美识
2026-04-27 16:49:50
港股交易时段,光通信龙头突发财报,股价瞬间飙10%,是否违规?

港股交易时段,光通信龙头突发财报,股价瞬间飙10%,是否违规?

金石随笔
2026-04-28 09:16:37
埃及、印尼、泰国接连放弃!歼-10CE越优秀,为何越难卖?

埃及、印尼、泰国接连放弃!歼-10CE越优秀,为何越难卖?

Ck的蜜糖
2026-04-28 19:07:41
750万“电鸡”压城:深圳“史上最严”整治仍面临五大难题

750万“电鸡”压城:深圳“史上最严”整治仍面临五大难题

南方都市报
2026-04-28 07:00:37
“崩老头”现象:00后如何收割中年男性

“崩老头”现象:00后如何收割中年男性

流苏晚晴
2026-04-25 16:31:35
克利夫兰媒体:骑士队愈发后悔交易得到哈登,已是彻头彻尾的败笔

克利夫兰媒体:骑士队愈发后悔交易得到哈登,已是彻头彻尾的败笔

宝哥精彩赛事
2026-04-28 17:49:58
名记:杜兰特或已打完火箭生涯最后一场球,下家是纽约和热火

名记:杜兰特或已打完火箭生涯最后一场球,下家是纽约和热火

懂球帝
2026-04-28 09:32:05
一个正处级干部退休后的真实生活:落差比想象中大

一个正处级干部退休后的真实生活:落差比想象中大

细说职场
2026-04-27 14:11:52
每斤降价30元左右!苏州河虾价格“跳水”倒计时,五一后迎来尝鲜最佳时期

每斤降价30元左右!苏州河虾价格“跳水”倒计时,五一后迎来尝鲜最佳时期

扬子晚报
2026-04-28 18:14:56
2026-04-28 19:43:00
机器之心Pro incentive-icons
机器之心Pro
专业的人工智能媒体
12881文章数 142638关注度
往期回顾 全部

科技要闻

10亿周活目标落空!传OpenAI爆发内部分歧

头条要闻

15岁女儿遭同学性侵杀害 父亲终日借酒浇愁饮酒300斤

头条要闻

15岁女儿遭同学性侵杀害 父亲终日借酒浇愁饮酒300斤

体育要闻

季后赛最新局势:雷霆4-0晋级首队 4队3-1

娱乐要闻

蔡卓妍官宣结婚,老公比她小10岁

财经要闻

政治局会议:加强算力网等规划建设

汽车要闻

拒绝疯狂套娃!现代艾尼氪金星长在未来审美点上

态度原创

房产
健康
时尚
公开课
军事航空

房产要闻

红利爆发!海南,冲到全国人口增量第4省!

干细胞治疗烧烫伤三大优势!

解锁刘浩存的「红运」密码 | 高清镜头后的底气

公开课

李玫瑾:为什么性格比能力更重要?

军事要闻

德国总理默茨:美国正遭受伊朗领导层的羞辱

无障碍浏览 进入关怀版