原生PyTorch支持,大模型一键迁移!寒武纪开源Torch-MLU(transformer pytorch源码)

AIGC动态欢迎阅读

原标题:原生PyTorch支持,大模型一键迁移!寒武纪开源Torch-MLU

关键字:开发者,设备,社区,插件,版本

文章来源:智东西

内容字数:0字

内容摘要:

寒武纪开源了Torch-MLU,并实现了寒武纪硬件对于PyTorch的原生支持。

今日,寒武纪开源了PyTorch设备后端扩展插件Torch-MLU,并实现了寒武纪硬件对于PyTorch的原生支持,充分提升了开发者的使用体验和集成效率。

早在2018年,寒武纪就开始了与PyTorch框架的集成适配工作。自PyTorch 1.3版本开始就支持寒武纪MLU系列智能加速卡作为PyTorch的加速后端,使开发者能够将原先基于GPU的深度学习网络高效迁移到基于寒武纪MLU的PyTorch环境中。

随着PyTorch社区发布PyTorch 2.4版本,配合新版本(≥ V1.22)的Torch-MLU插件,寒武纪硬件实现了对PyTorch的原生支持。开发者无需再安装寒武纪定制的Cambricon PyTorch,而是可以基于社区的PyTorch安装Torch-MLU插件,就可将寒武纪硬件的算力接入原生PyTorch。

与此同时,Torch-MLU的开源使寒武纪能够更快地响应开发者的反馈和问题,搭建了高效的沟通桥梁,促进了与全球开发者的协作和知识共享。寒武纪通过持续贡献开源社区的实际行动,展示着拥抱开

原文链接:原生PyTorch支持,大模型一键迁移!寒武纪开源Torch-MLU

联系作者

文章来源:智东西

作者微信:

作者简介:

0
分享到:
没有账号? 忘记密码?