init_rpc

paddle.distributed.rpc. init_rpc ( name, rank=None, world_size=None, master_endpoint=None )

初始化 rpc 服务。注意:所有 RPC 对外开放接口,均应在安全的网络环境下供内部使用,不得公网开放访问。

参数

  • name (str) - worker 名字。

  • rank (int,可选) - worker 的 ID,默认为 None。

  • world_size (int,可选) - workers 的数量,默认为 None。

  • master_endpoint (str,可选) - master 的 IP 地址,其他 worker 节点通过 master 来获取彼此的信息,默认是 None。

返回

代码示例

COPY-FROM: paddle.distributed.rpc.init_rpc