mmengine.dist.init_dist — mmengine 0.11.0rc0 documentation (original) (raw)
- Docs >
- mmengine.dist >
- mmengine.dist.init_dist
- 以中文阅读
Shortcuts
mmengine.dist.init_dist(launcher, backend='nccl', init_backend='torch', **kwargs)[source]¶
Initialize distributed environment.
Parameters:
- launcher (str) – Way to launcher multi processes. Supported launchers are ‘pytorch’, ‘mpi’ and ‘slurm’.
- backend (str) – Communication Backends. Supported backends are ‘nccl’, ‘gloo’ and ‘mpi’. Defaults to ‘nccl’.
- **kwargs – keyword arguments are passed to
init_process_group.
Return type:
None