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

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

相关推荐
热点推荐
牢A警告留学生家长:不要把你们的老婆,送到欧美社会去陪读

牢A警告留学生家长:不要把你们的老婆,送到欧美社会去陪读

番茄说史聊
2026-01-20 10:53:58
皮尔斯:当今NBA只有四个人算得上超巨,詹杜库都不够格

皮尔斯:当今NBA只有四个人算得上超巨,詹杜库都不够格

漫川舟船
2026-01-22 16:26:15
-16.5℃!临沂最低温!接下来雨夹雪或小到中雪!

-16.5℃!临沂最低温!接下来雨夹雪或小到中雪!

琅琊新闻网
2026-01-22 15:37:56
老人 “黄金糖化值” 出炉!不是6.0,这个数才对,越接近越放心

老人 “黄金糖化值” 出炉!不是6.0,这个数才对,越接近越放心

垚垚分享健康
2026-01-21 12:45:07
中国股市:但凡涨停股,如果出现这两个信号,就是强势连板大牛股

中国股市:但凡涨停股,如果出现这两个信号,就是强势连板大牛股

一方聊市
2026-01-22 13:48:47
美国首次击落一架光纤无人机

美国首次击落一架光纤无人机

老马拉车莫少装
2026-01-21 17:38:49
当亚历山大19中16轰40+11+7时,你才会发现张伯伦的可怕之处

当亚历山大19中16轰40+11+7时,你才会发现张伯伦的可怕之处

篮球大视野
2026-01-22 15:32:13
“点都德”老字号的迷局,为什么老广州不买账?四点原因说真相

“点都德”老字号的迷局,为什么老广州不买账?四点原因说真相

珊姐姐
2026-01-21 10:02:01
郑丽文立一件大功?黄国昌从美国返程后,对大陆态度有了罕见变化

郑丽文立一件大功?黄国昌从美国返程后,对大陆态度有了罕见变化

老鹈爱说事
2026-01-22 11:41:38
某涉密单位办公室搬迁,两台打印机被拉到废品回收站,文印室管理员工、实习生等被严肃处理

某涉密单位办公室搬迁,两台打印机被拉到废品回收站,文印室管理员工、实习生等被严肃处理

都市快报橙柿互动
2026-01-21 07:55:22
涉烟花爆竹,郑州多人被拘留

涉烟花爆竹,郑州多人被拘留

大象新闻
2026-01-22 13:17:09
难怪扈三娘不恨宋江,你看宋江擒下她那晚,夺走了她的什么就懂了

难怪扈三娘不恨宋江,你看宋江擒下她那晚,夺走了她的什么就懂了

芊芊子吟
2026-01-16 22:25:03
昔年混入红军的土匪头子,建国后毛主席指示贺晋年:此人一定要抓

昔年混入红军的土匪头子,建国后毛主席指示贺晋年:此人一定要抓

鹤羽说个事
2026-01-22 11:51:54
加油被拒将油呲地上,加油员崩溃大哭 男子正脸被扒已社死 恐被刑拘

加油被拒将油呲地上,加油员崩溃大哭 男子正脸被扒已社死 恐被刑拘

特特农村生活
2026-01-22 13:34:08
赞不绝口,勇士队老板对库明加的“咆哮式反应”引爆NBA圈内

赞不绝口,勇士队老板对库明加的“咆哮式反应”引爆NBA圈内

好火子
2026-01-22 05:04:55
庐山会议后,警卫参谋景希珍对彭总不离不弃,他的最终结局如何?

庐山会议后,警卫参谋景希珍对彭总不离不弃,他的最终结局如何?

鹤羽说个事
2026-01-22 11:42:32
妻子照顾瘫痪公公13年,丈夫提离婚公公同意,可走出大门丈夫懵了

妻子照顾瘫痪公公13年,丈夫提离婚公公同意,可走出大门丈夫懵了

兰姐说故事
2026-01-15 20:30:04
WTT常规赛:林诗栋开门红!首局2平后拉开分差,对手4-11惨败!

WTT常规赛:林诗栋开门红!首局2平后拉开分差,对手4-11惨败!

刘姚尧的文字城堡
2026-01-22 16:03:50
官宣:北京成为大陆第二个5万亿GDP城市,有望近两年赶超日本东京

官宣:北京成为大陆第二个5万亿GDP城市,有望近两年赶超日本东京

户外钓鱼哥阿旱
2026-01-22 00:05:40
女子拍摄姑姑偷玉米被姑父棒杀!姑姑哀求出谅解书,女儿绝不原谅

女子拍摄姑姑偷玉米被姑父棒杀!姑姑哀求出谅解书,女儿绝不原谅

叶公子
2026-01-21 18:50:53
2026-01-22 17:03:00
机器之心Pro incentive-icons
机器之心Pro
专业的人工智能媒体
12168文章数 142547关注度
往期回顾 全部

科技要闻

几千亿只是开胃菜,AI基建还得再砸几万亿

头条要闻

浙江有人领走88888888马年纪念钞 银行员工:随机分配

头条要闻

浙江有人领走88888888马年纪念钞 银行员工:随机分配

体育要闻

跑个步而已,他们在燃什么?

娱乐要闻

钟丽缇土耳其高空落泪 与张伦硕拥吻

财经要闻

申通快递创始人被前夫索要股份

汽车要闻

配备多块娱乐屏 极氪8X内饰曝光

态度原创

本地
数码
家居
时尚
亲子

本地新闻

云游中国|格尔木的四季朋友圈,张张值得你点赞

数码要闻

国补叠加限时直降,这篇华为平板上车攻略不容错过

家居要闻

法式风情 南洋中古居

50+女性穿衣没头绪?教你3个显瘦还时髦的思路,照搬就好看

亲子要闻

家长因孩子发烧,深夜带老师cosplay“大萨满”,评论区太精彩了

无障碍浏览 进入关怀版