← 返回事件
持续讨论AI发版

viable/strict/1790620535: [c10d] Allow CPU device_id in init_process_group (#198340)

图:PyTorch Releases

发生了什么

_new_process_group_helper rejected any CPU device_id, even one with a valid index (e.g. cpu:0), raising ValueError: init_process_group device_id parameter must be an accelerator with an index. This made it impossible to use device_id with the GLOO backend on CPU, despite init_process_group itself already treating CPU device_id as a valid case (it skips the accelerator-matching checks specifically for CPU, at line ~2…

摘要按规则整理自下方来源原文

为什么在扩散

来源