一，前言

在深度學習框架的不斷迭代中，PyTorch 社區(qū)始終致力于提供更穩(wěn)定、更高效的工具。最近，PyTorch 2.5.1 版本正式發(fā)布，這個版本主要針對 2.5.0 中發(fā)現(xiàn)的問題進行了修復，以提升用戶體驗。

二，PyTorch 2.5.1 的主要修內(nèi)容

1，RPM 和 arm64 發(fā)行版支持：

2.5.1 版本修復了基于 RPM 的發(fā)行版和 arm64 發(fā)行版中的一些問題，這些修復使得在這些系統(tǒng)上運行 PyTorch 變得更加穩(wěn)定。

2，torch.compile 崩潰問題：

針對 torch.compile 過程中出現(xiàn)的崩潰問題，2.5.1 版本提供了解決方案，這對于使用 PyTorch 進行模型編譯的用戶來說是一個重要的更新。

3，MPS 崩潰修復：

在 2.5.0 版本中，Metal Performance Shaders (MPS) 的使用出現(xiàn)了一些崩潰問題，2.5.1 版本對這些問題進行了修復，提升了在 iOS 和 macOS 上的性能。

4，注意力回歸問題：

2.5.0 版本中觀察到的注意力回歸問題也在 2.5.1 中得到了解決，這對于使用注意力機制的模型來說是一個重要的改進。

三，PyTorch?2.5.1 版本的其他改進

除了修復錯誤，2.5.1 版本還帶來了一些其他的改進，包括對第三方設備支持的增強，以及對 timm 和 OpenCLIP 的更新，以更好地支持設備自動加載擴展。

如何獲取 PyTorch 2.5.1

用戶可以通過 PyTorch 官方網(wǎng)站或通過 pip 安裝 PyTorch 2.5.1 版本。對于想要了解更多關于這個版本修復內(nèi)容的用戶，可以訪問 PyTorch 官方發(fā)布筆記。

四，總結

PyTorch 2.5.1 版本的發(fā)布，展示了 PyTorch 社區(qū)對穩(wěn)定性和性能的持續(xù)承諾。這個修復版對于那些在 2.5.0 版本中遇到問題的用戶來說是一個福音。社區(qū)鼓勵所有 PyTorch 用戶升級到這個新版本，以享受更流暢的深度學習體驗。如果大家在新版本中遇到任何問題，PyTorch 社區(qū)也歡迎大家在 GitHub 上提出，以便持續(xù)改進和優(yōu)化框架。通過這樣的互動，PyTorch 能夠不斷進步，滿足不斷增長的深度學習研究和開發(fā)需求。

如果你有更好的文章，歡迎投稿！

稿件接收郵箱：nami.liu@pasuntech.com

更多精彩內(nèi)容請關注“算力魔方?”！