跳到正文
PyTorch Blog(pytorch.org)· PyTorch Foundation·· 2 天前精選品質評分85

PyTorch 2.15 釋出全新 NCCL 後端、MPS FP8 與跨平臺最佳化

PyTorch 2.15 Release Blog

重點導讀

PyTorch 2.15 釋出全新 NCCL 後端、MPS FP8 支援以及跨平臺最佳化,為多 GPU 分散式訓練與低精度推理帶來顯著效能提升。

  • NCCL:預設使用重寫後端,並提供 TORCH_DIST_USE_NCCL2=0 或 nccl-legacy 作為回滾。
推薦理由

PyTorch 2.15 針對多平臺、分散式訓練與低精度推理進行大幅最佳化,並加入全新 NCCL 後端與 MPS FP8 支援,開發者可即時驗證新功能提升效能,同時提供 CUDA 13.2/13.4 以及 ROCm 10.0 wheels,讓跨平臺部署更順暢。

信源出處pytorch.org

PyTorch 2.15 Release Blog

本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。

開啟原文
來源: PyTorch Blog (pytorch.org)
開啟原文