Add a specific device_id when calling torch.init_process_group to avoid NCCL communication errors. - #76
Open
gongweibao wants to merge 1 commit into
Open
Add a specific device_id when calling torch.init_process_group to avoid NCCL communication errors.#76gongweibao wants to merge 1 commit into
device_id when calling torch.init_process_group to avoid NCCL communication errors.#76gongweibao wants to merge 1 commit into
Commits
Commits on Mar 5, 2025
- committed
