PyTorch 2.15 发布:重写 NCCL 后端成为默认,Apple Silicon 支持 FP8
PyTorch 基金会发布 PyTorch 2.15,所有使用 nccl 的进程组默认改用重写后的 NCCL 后端,并附带在集合通信挂起时转储现场信息的 flight recorder,可通过 TORCH_DIST_USE_NCCL2=0 或 nccl-legacy 回退。该版本还为 Apple Silicon 加入 float8_e4m3fn 与 _scaled_mm 支持,新的 grouped_mm 在 M3 Ultra 上让 MoE 解码吞吐提升 1.32–1.34 倍。默认 wheel 升级到 CUDA 13.2,同时提供 CUDA 13.4 wheel 并实验性支持 NVIDIA Rubin GPU,另新增 ROCm 10.0 wheel,且不再支持 Python 3.10 和 macOS 14。
#产品更新#开源生态#部署/工程



