【ITBEAR】摩尔线程近日宣布,其针对PyTorch深度学习框架的MUSA插件——Torch-MUSA,已推出新版本v1.3.0。此次更新全面兼容PyTorch 2.2.0,不仅提升了模型在MUSA架构上的性能与覆盖度,更支持模型轻松迁移到国产全功能GPU上运行。
Torch-MUSA插件的推出,为PyTorch用户提供了便捷的MUSA后端加速支持。用户只需简单指定设备为"musa",便可在MUSA架构上高效运行深度学习模型,充分发挥国产GPU的计算潜能。
据悉,PyTorch作为广泛应用于自然语言处理、计算机视觉等领域的深度学习框架,其模型运行效率一直备受关注。而摩尔线程通过不断优化Torch-MUSA,旨在为用户提供更加流畅、高效的深度学习体验。
新版本v1.3.0的发布,标志着Torch-MUSA在兼容性和性能上又迈出了重要一步。未来,摩尔线程将继续跟进PyTorch的版本更新,为用户提供更多先进功能和更佳性能支持。
PyTorch旗下架构优化库torchao的发布,也为深度学习模型的量化和稀疏性优化提供了有力工具。这一优化库能够在保持性能的同时,有效降低模型计算成本和RAM用量,与Torch-MUSA的更新相得益彰。