PyTorch 2.15发布并重写NCCL后端

发生了什么

据The Register报道，PyTorch 2.15版本正式发布，核心更新包括重写Nvidia NCCL后端以优化故障诊断与恢复机制。新版NCCL后端通过限制每个进程组仅使用一个通信器，旨在减少死锁并提升性能。此外，该版本新增“飞行记录器”功能以捕获故障快照，并支持Nvidia Blackwell GPU及Apple设备的并行处理改进。旧版NCCL后端作为回滚选项保留，计划于2.16版本中移除。