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

一文详述华为最新开源框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年一季度全国无一省份实现财政自平衡

史上首次!2026年一季度全国无一省份实现财政自平衡

大江看潮
2026-07-09 08:13:10
美出重拳,越5天8小时工作制,中资从越南撤退,新世界工厂在我国中西部

美出重拳,越5天8小时工作制,中资从越南撤退,新世界工厂在我国中西部

王新喜
2026-07-10 22:11:39
曲波:当年留洋我连脚指甲都不能自己剪;走出去后要积极融入

曲波:当年留洋我连脚指甲都不能自己剪;走出去后要积极融入

懂球帝
2026-07-11 04:50:13
灾难事件中的歌颂叙事,越来越让人反感

灾难事件中的歌颂叙事,越来越让人反感

黔有虎
2026-07-10 17:13:47
国内学者研讨声索巴丹群岛主权

国内学者研讨声索巴丹群岛主权

环球网资讯
2026-07-10 12:10:45
24岁比利时门将崩溃:替补出场后扑救脱手 西班牙神兵连续2场绝杀

24岁比利时门将崩溃:替补出场后扑救脱手 西班牙神兵连续2场绝杀

风过乡
2026-07-11 05:03:44
花蚊子入侵中国五十年,人们夜晚再也不能乘凉,它到底来自哪里?

花蚊子入侵中国五十年,人们夜晚再也不能乘凉,它到底来自哪里?

铭记历史呀
2026-07-10 19:30:52
每次有灾情公司像过年一样?网友曝韩红基金会内幕:在自家公司进货,最高加价180%

每次有灾情公司像过年一样?网友曝韩红基金会内幕:在自家公司进货,最高加价180%

爆角追踪
2026-07-10 07:52:07
全面禁售燃油车第一省确定,全国实施还有多远

全面禁售燃油车第一省确定,全国实施还有多远

Mr王的饭后茶
2026-07-10 21:54:46
管水库的人因承包出去养鱼,不舍得放水致水库漫堤?六蓝水库养殖公告照片引争议

管水库的人因承包出去养鱼,不舍得放水致水库漫堤?六蓝水库养殖公告照片引争议

火山詩话
2026-07-07 11:02:52
200万粉丝博主称向广西捐款2万元被网暴,被质疑金额太少,还有人私信她“会遭报应”,当事人回应

200万粉丝博主称向广西捐款2万元被网暴,被质疑金额太少,还有人私信她“会遭报应”,当事人回应

都市快报橙柿互动
2026-07-11 03:51:48
突发!香港知名演员被TVB和邵氏同时除名,深陷偷高层礼物风波

突发!香港知名演员被TVB和邵氏同时除名,深陷偷高层礼物风波

一盅情怀
2026-07-10 15:53:53
“飞行员举报情人诈骗700余万案”二审改判:刑期从12年减为6年,家属称将申诉

“飞行员举报情人诈骗700余万案”二审改判:刑期从12年减为6年,家属称将申诉

红星新闻
2026-07-10 18:31:31
晋江鞋厂大火:你以为的廉价,是别人的代价

晋江鞋厂大火:你以为的廉价,是别人的代价

木蹊说
2026-07-10 15:52:53
悲情谢幕!比利时黄金一代退出舞台:12年0冠 曾长期排名世界第1

悲情谢幕!比利时黄金一代退出舞台:12年0冠 曾长期排名世界第1

风过乡
2026-07-11 05:56:13
前队友爆发冲突!阿德巴约动手打希罗 疑似不满对方吐槽自己不值顶薪

前队友爆发冲突!阿德巴约动手打希罗 疑似不满对方吐槽自己不值顶薪

罗说NBA
2026-07-11 05:52:00
世界杯很可能历史重演,法国第3,阿根廷第2,新军登顶夺冠

世界杯很可能历史重演,法国第3,阿根廷第2,新军登顶夺冠

铁甲西奇
2026-07-10 10:02:26
直落三盘横扫晋级!辛纳终结德约25冠梦,第7次跻身大满贯决赛

直落三盘横扫晋级!辛纳终结德约25冠梦,第7次跻身大满贯决赛

全景体育V
2026-07-11 03:48:12
要做好“潮水退却”的准备,房价或将超出想象

要做好“潮水退却”的准备,房价或将超出想象

麓谷隐士
2026-07-10 00:45:03
库瓦在场下看傻了!24岁曼联门神巨大失误坑惨黄金一代 卢卡库呆立

库瓦在场下看傻了!24岁曼联门神巨大失误坑惨黄金一代 卢卡库呆立

风过乡
2026-07-11 05:40:58
2026-07-11 07:31:00
机器之心Pro incentive-icons
机器之心Pro
专业的人工智能媒体
13481文章数 142690关注度
往期回顾 全部

科技要闻

中国开启可回收火箭时代

头条要闻

世界杯-西班牙2-1比利时将战法国 梅里诺连场绝杀

头条要闻

世界杯-西班牙2-1比利时将战法国 梅里诺连场绝杀

体育要闻

法国VS摩洛哥:谁才是臭外地的?

娱乐要闻

韩国顶流李钟硕与IU官宣分手!

财经要闻

一封举报信 引发小红书IPO合规考验

汽车要闻

吉利银河TT:C级纯电轿跑新玩家 此TT非彼TT

态度原创

房产
家居
时尚
艺术
亲子

房产要闻

重磅学校规划曝光!西海岸教育,正强得可怕!

家居要闻

2026建博会(广州) 公装联探展交流活动

夏天真的很适合穿黄色,比黑色减龄、比白色时尚,高级显元气

艺术要闻

陈子庄『花卉禽鸟册』

亲子要闻

儿子跟女儿都好

无障碍浏览 进入关怀版