跳到正文
PyTorch Blog· PyTorch Foundation·· 2 小时前精选AI 评分62

PyTorch 2.15 发布:重写版 NCCL 后端成为默认

PyTorch 2.15 Release Blog

AI 导读

PyTorch 2.15 发布,重写版 NCCL 后端成为所有 nccl 进程组的默认,并提供 TORCH_DIST_USE_NCCL2=0 与 nccl-legacy 回滚路径。

推荐理由

官方发布说明集中列出分布式训练、低精度解码和多硬件后端的变化,便于对照现有训练栈评估升级影响。

来源:PyTorch Blog · pytorch.org