mmengine.dist.init_local_group¶
- mmengine.dist.init_local_group(node_rank, num_gpus_per_node)[源代码]¶
Setup the local process group.
Setup a process group which only includes processes that on the same machine as the current process.
The code is modified from https://github.com/facebookresearch/detectron2/blob/main/detectron2/engine/launch.py