发射器(launcher)#

class launcher(*args, **kwargs)[源代码]#

Decorator for launching multiple processes in single-machine multi-gpu training.

参数:
  • func – 你想要在分布式模式下启动的函数。

  • n_gpus – how many devices each node.

  • world_size – how many devices totally.

  • rank_start – start number for rank.

  • master_ip – ip address for master node (where the rank 0 is).

  • port – server port for distributed server.

  • backend – set default collective communication backend.