沃资讯
科技 财经 汽车 游戏 数码 资讯

摩尔线程Torch-MUSA插件v1.3.0开源升级,加速替代CUDA的步伐!

2024-10-31来源:ITBEAR编辑:瑞雪

摩尔线程近日宣布,其针对PyTorch深度学习框架的MUSA插件“Torch-MUSA”已推出重大更新的v1.3.0版本,该版本全面兼容PyTorch 2.2.0,进一步提升了模型在摩尔线程GPU上的性能与覆盖度。

作为全球广泛应用的深度学习框架,PyTorch在自然语言处理、计算机视觉等领域具有强大的实力。摩尔线程的Torch-MUSA插件为PyTorch用户提供了MUSA后端加速,使得深度学习模型能够在MUSA架构上更加流畅地运行。

自Torch-MUSA首次发布以来,已经历了多个版本的迭代升级,不断优化兼容性与性能。从v1.0.0版本开始,该插件就率先支持了PyTorch 2.0,为用户带来了显著的计算加速与更好的易用性。

最新发布的v1.3.0版本除了全面支持PyTorch 2.2.0外,还极大地提升了模型训练与推理的效率,更好地满足了高性能深度学习任务的需求。该版本还增加了对FSDP等特性的支持,使得用户能够处理更复杂的模型和更大规模的数据。

目前,Torch-MUSA项目已完全开源,开发者可通过GitHub获取源代码,并积极参与项目的开发与改进。摩尔线程鼓励开发者通过提交问题报告或代码修改申请等方式,共同推动Torch-MUSA及MUSA软件生态的持续发展。

Torch-MUSA的功能特性包括轻松迁移现有的PyTorch模型到MUSA架构GPU上运行,完全兼容PyTorch的自动微分与动态图机制,并支持多种神经网络模块、优化算法及关键深度学习算子的加速。该插件还支持多种PyTorch特性,为用户提供了更为丰富的功能选择。

随着深度学习技术的不断发展,摩尔线程将继续跟进PyTorch的版本更新,并计划支持更高版本的PyTorch,以满足用户不断增长的需求。

奥迪A7内饰大揭秘:新能源风格设计,智能化再升级!
奥迪近期对其车型命名规则进行了调整,奇数编号车型(如A3、A5、A7)将继续配备传统内燃机,而偶数编号车型(如A4、A6、A8)将与“e-tron”标识结合,成为纯电动汽车的专属命名方式。这一策略在近期发布的…

2024-10-31

MPV卫冕之战 别克推出全新GL8陆尊
新能源呼声越来越高,但就1-10月高端MPV市场的销量来看,以丰田塞纳/格瑞维亚、传祺M8为代表的传统燃油MPV,依然占据了绝对优势。针对这一市场变化,10月31日,上汽通用汽车别克品牌宣布全新GL8陆尊正式…

2024-10-31