init_rpc

paddle.distributed.rpc. init_rpc ( name, rank=None, world_size=None, master_endpoint=None ) [源代码]

初始化 rpc 服务。

参数

  • name (str) - worker 名字。

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

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

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

返回

代码示例

import paddle.distributed.rpc as rpc

rpc.init_rpc("worker0", rank=0, world_size=1,
            master_endpoint="127.0.0.1:8001")
rpc.shutdown()