?一，前言

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

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

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

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

2，torch.compile 崩潰問題：

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

3，MPS 崩潰修復(fù)：

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

4，注意力回歸問題：

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

三，PyTorch2.5.1 版本的其他改進(jìn)

除了修復(fù)錯(cuò)誤，2.5.1 版本還帶來了一些其他的改進(jìn)，包括對(duì)第三方設(shè)備支持的增強(qiáng)，以及對(duì) timm 和 OpenCLIP 的更新，以更好地支持設(shè)備自動(dòng)加載擴(kuò)展。

如何獲取 PyTorch 2.5.1

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

四，總結(jié)

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

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

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

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

?
審核編輯 黃宇