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

一文详述华为最新开源框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.

相关推荐
热点推荐
反转!巴西部长强行封杀比亚迪,卢拉深夜震怒:立刻卷铺盖走人!

反转!巴西部长强行封杀比亚迪,卢拉深夜震怒:立刻卷铺盖走人!

共工之锚
2026-04-15 00:31:36
郑丽文和家人的一张合影,时间是1971年,地点是眷村。

郑丽文和家人的一张合影,时间是1971年,地点是眷村。

叶老四
2026-04-15 09:44:36
一杯茶蒸发300亿!霸王茶姬泡沫破裂,创始人卷7亿走人

一杯茶蒸发300亿!霸王茶姬泡沫破裂,创始人卷7亿走人

老特有话说
2026-04-14 13:40:08
恒大夏海钧:当一个人高智商,有资源,无下限,会坏到什么地步?

恒大夏海钧:当一个人高智商,有资源,无下限,会坏到什么地步?

道术意义
2026-04-08 07:32:08
国台办:坚决反对建交国与中国台湾地区开展任何形式官方往来

国台办:坚决反对建交国与中国台湾地区开展任何形式官方往来

环球网资讯
2026-04-15 10:38:12
54岁班主任带15岁女孩到宾馆开房猥亵被拘10天,家属发声

54岁班主任带15岁女孩到宾馆开房猥亵被拘10天,家属发声

澎湃新闻
2026-04-14 22:03:12
河南13岁被锁喉男孩赴京治疗后已脱离生命危险,拉绳老人已被采取刑事强制措施

河南13岁被锁喉男孩赴京治疗后已脱离生命危险,拉绳老人已被采取刑事强制措施

极目新闻
2026-04-15 11:11:27
32岁丹麦羽毛球名将安赛龙自宣退役:我的身体已经无法支撑我继续作战

32岁丹麦羽毛球名将安赛龙自宣退役:我的身体已经无法支撑我继续作战

懂球帝
2026-04-15 13:28:18
英国、日本、加拿大、巴西、瑞士、约旦、澳大利亚、哥伦比亚、印度尼西亚、塞拉利昂,发表联合声明

英国、日本、加拿大、巴西、瑞士、约旦、澳大利亚、哥伦比亚、印度尼西亚、塞拉利昂,发表联合声明

新京报政事儿
2026-04-15 11:41:04
企业增值税的设计:亏损了也要交税

企业增值税的设计:亏损了也要交税

生命可以承受之轻
2026-04-15 06:48:23
这居然是任泉?网友:真认不出来了

这居然是任泉?网友:真认不出来了

木子爱娱乐大号
2026-04-14 10:52:36
要想拿到20亿的遗产,得交11亿的税?中山美穗独子放弃继承遗产引发日本社会关注…

要想拿到20亿的遗产,得交11亿的税?中山美穗独子放弃继承遗产引发日本社会关注…

日本通
2026-04-14 15:03:49
裁判回应争议!三球拉拽阿德巴约不能回看 后续联盟决定是否升级

裁判回应争议!三球拉拽阿德巴约不能回看 后续联盟决定是否升级

罗说NBA
2026-04-15 11:44:51
这8种病立马办残疾证!符合条件每月领钱,别傻傻把福利扔了!

这8种病立马办残疾证!符合条件每月领钱,别傻傻把福利扔了!

记录生活日常阿蜴
2026-04-14 07:01:50
央视曝光:举债3.68亿、总投资超7亿的“农业实训基地”,只有会堂、酒店、健身房、棋牌室等,找不到一处跟种植、养殖等相关的场所和设备

央视曝光:举债3.68亿、总投资超7亿的“农业实训基地”,只有会堂、酒店、健身房、棋牌室等,找不到一处跟种植、养殖等相关的场所和设备

扬子晚报
2026-04-14 22:36:32
美伊谈判前景堪忧?美国对华恩将仇报,不许中方出手帮伊朗

美伊谈判前景堪忧?美国对华恩将仇报,不许中方出手帮伊朗

第一军情
2026-04-15 11:00:03
许家印,KTV豪情终成妄念

许家印,KTV豪情终成妄念

哲空空
2026-04-15 11:40:51
福州一河流现大量巴西龟,岸边留塑料筐,疑似有人倾倒,管理方:将调取监控找人;律师:丢弃外来入侵物种可处一万元以上五万元以下罚款

福州一河流现大量巴西龟,岸边留塑料筐,疑似有人倾倒,管理方:将调取监控找人;律师:丢弃外来入侵物种可处一万元以上五万元以下罚款

大风新闻
2026-04-15 11:47:05
陈根:美国正式封锁霍尔木兹海峡,为什么中国是最受伤的一方?

陈根:美国正式封锁霍尔木兹海峡,为什么中国是最受伤的一方?

陈述智能
2026-04-15 07:45:08
郑丽文愿与赖清德沟通但未收到回应,国台办引用《论语》评论

郑丽文愿与赖清德沟通但未收到回应,国台办引用《论语》评论

澎湃新闻
2026-04-15 11:12:26
2026-04-15 14:03:00
机器之心Pro incentive-icons
机器之心Pro
专业的人工智能媒体
12767文章数 142627关注度
往期回顾 全部

科技要闻

手机无死角上网?亚马逊砸百亿硬刚马斯克

头条要闻

上万美军封锁下 多艘货船成功通过霍尔木兹海峡

头条要闻

上万美军封锁下 多艘货船成功通过霍尔木兹海峡

体育要闻

三球准绝杀戴大金链:轰30+10自我救赎

娱乐要闻

曾志伟办73岁生日派对,逾百艺人到场

财经要闻

业绩失速的Lululemon:"健康"人设崩塌?

汽车要闻

海豹08内饰首秀 大满配“海王”旗舰

态度原创

房产
时尚
旅游
本地
家居

房产要闻

重磅调规!341亩商改住+中小学用地!宝龙城这把稳了?

坏事做尽的疯女人,集体翻红了

旅游要闻

国色天香!盐城珠溪古镇多处牡丹绽放

本地新闻

12吨巧克力有难,全网化身超级侦探添乱

家居要闻

简而不减 暖居之道

无障碍浏览 进入关怀版