掘金 人工智能 前天 17:49
Day0 迁移、一键部署,华为开源的昇思 MindSpore 成为大模型开发的 “万能钥匙”
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

面对众多AI大模型和技术,开发者面临着迁移和部署的挑战。昇思 MindSpore提供了快速、便捷的解决方案。通过MSAdapter,开发者只需少量改动代码即可迁移主流大模型,并实现一键部署。MindSpore支持训练Day0迁移,推理一键部署,大幅降低了开发者的工作量。例如,DeepSeek-V3的迁移只需修改少量代码,推理部署也只需半小时。MindSpore还通过多项技术优化,提升了训练和推理的性能,例如JIT编译、自动策略寻优等。

🚀 昇思 MindSpore 提供了快速迁移大模型的解决方案,核心在于其“翻译神器”——MSAdapter。该工具能够将其他框架的代码转换为MindSpore可识别的语言,实现近乎零损耗的迁移,例如PyTorch代码可以直接在MindSpore上运行。

💡 MindSpore 通过多项技术加速训练和调试调优,包括动态图多级流水、JIT编译、自动策略寻优和执行序比对。这些技术提升了训练速度和效率,例如在DeepSeek-V3的训练中,性能提升了9.5%。

🔑 在推理部署方面,MindSpore 结合vLLM-MindSpore插件,使得HuggingFace模型能够在半小时内完成部署。它支持直接加载HuggingFace的模型权重,并提供三层部署模式,减少了启动延迟。例如,百亿参数模型加载时间缩短至30秒以内,Pangu Pro MoE 72B模型在Atlas 800I A2上实现了每秒1020tokens的吞吐量。

没有一个大模型可以一统天下。

这,或许已经成为了 AI 大模型时代行业里的一个共识。

在如此背景之下,面对众多且日新月异的主流大模型和 AI 技术,如何能在一个框架、生态下去体验,却成了开发者们 “老大难” 的问题。

难道就没有一个又快又好又方便的解决办法吗?

有的——

华为开源的昇思 MindSpore,了解一下。

在这里,主流 SOTA 大模型的 “搬家” 是这样的——训练 Day0 迁移

只需改动极少极少的代码就 OK,并且精度和性能都在线。

推理一键部署的:

训练转推理全流程自动化,20 多个主流大模型开箱即用,百亿参数模型加载只需不到 30 秒。

**

**

MindSpore 生态快速迁移解决方案的技术架构

那么昇思 MindSpore 是如何做到,我们继续往下看。

改 4 行代码,“搬家”DeepSeek-V3

为了让迁移大模型这件事变得无感知,昇思 MindSpore“翻译神器”——MSAdapter

简单来说,这个工具可以把其他框架的代码转换成 MindSpore 能看懂的语言,从而实现 “零损耗” 迁移。

比如 PyTorch 写的训练脚本,直接在 MindSpore 里运行,动态图体验和原来一样顺手,95% 以上的接口都能自动转换,迁移损耗几乎为零。

在此背后还有其他的 “独家秘笈”,加速训练调试调优,具体技术如下:

**

**

MindSpore 训练 Day0 迁移方案

以 DeepSeek-V3 为例,代码改动量如下:

不仅精度上可以实现 Day0 对齐,而且在保持并行策略一致的情况下叠加 MindSpore 自研增量特性,性能还能提升 5%

HuggingFace 模型们,推理一键部署

在推理部署这块儿,昇思则是用 vLLM-MindSpore 插件能让 HuggingFace 模型在半小时内完成部署并上线。

**

**

vLLM x MindSpore 框架图

要是碰到千亿参数的大模型,MindSpore 用三层部署模式重新打造了推理流程:

从实测效果来看,以 Pangu Pro MoE 72B 为例,使用 vLLM 和 MindSpore 在 Atlas 800I A2 上部署推理服务,当前在时延小于 100ms 的情况下单卡增量吞吐可达每秒 1020tokens,在 Atlas 300I Pro 上可达每秒 130tokens。

以上便是关于昇思 MindSpore“训练 Day0 迁移、推理一键部署” 的大致内容了,了解更多详情可戳

技术报告地址:

gitcode.com/ascend-trib…

欢迎在评论区留下你的想法!

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

昇思MindSpore 大模型迁移 一键部署
相关文章