Shortcuts

mmengine.dist.init_dist

mmengine.dist.init_dist(launcher, backend='nccl', **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

Read the Docs v: v0.5.0
Versions
latest
stable
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.