跳到正文
掘金 · AI 编程· 知识搬运临时工·· 2 小时前AI 评分58

PyTorch 2.15 重写 NCCL 后端并加入挂住诊断的飞行记录仪

PyTorch 2.15 先把 NCCL 那笔债还了

AI 导读

PyTorch 2.15 于 10 月 28 日发布,把重写过的 NCCL 后端设为所有 nccl process group 的默认实现,该后端在 2.14 中还需手动开启。

来源:掘金 · AI 编程 · juejin.cn