💡
原文中文,约1400字,阅读约需4分钟。
📝
内容提要
摩尔线程发布了Torch-MUSA v1.3.0,全面兼容PyTorch 2.2.0,提升模型性能并支持国产GPU,用户可轻松迁移模型。Torch-MUSA开源,鼓励开发者参与改进,未来将持续跟进PyTorch更新。
🎯
关键要点
- 摩尔线程发布了Torch-MUSA v1.3.0,全面兼容PyTorch 2.2.0。
- 新版本提升了模型性能,支持国产全功能GPU,用户可轻松迁移模型。
- Torch-MUSA专为PyTorch提供MUSA后端加速支持,提升深度学习模型运行效率。
- Torch-MUSA已完全开源,鼓励开发者参与改进,提交问题报告或代码修改申请。
- 用户只需指定torch.device(“musa”)即可迁移现有PyTorch模型,无需大幅修改代码。
- Torch-MUSA支持多种PyTorch特性,包括DDP、JIT、FSDP、Profiler、Extension等。
- 版本迭代包括v1.1.0(支持PyTorch 2.0)、v1.2.0(扩展算子支持)、v1.3.0(支持PyTorch 2.2.0)。
- 未来Torch-MUSA将继续跟进PyTorch版本更新,计划支持更高版本。
➡️