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

PyTorch 2.15 发布,重写版 NCCL 后端成为默认并扩展多硬件支持

PyTorch 2.15 Release Blog

AI 导读

PyTorch 2.15 已发布,重写版 NCCL 后端现为所有 nccl process group 的默认后端,并在 collective 挂起时通过 flight recorder 将记录写入磁盘。

推荐理由

PyTorch 2.15 将重写版 NCCL 后端设为默认,并把更新扩展到分布式训练、低精度推理与多类硬件平台,呈现出本次版本的系统性变化。

来源:PyTorch Blog · pytorch.org