get_worker_info¶

paddle.distributed.rpc. get_worker_info ( name ) [源代码] ¶

利用 worker 名字获取 worker 的信息。

参数¶

name (str) - worker 的名字。

返回¶

WorkerInfo 对象，拥有属性 name，rank，ip，port。

代码示例¶

          >>> import paddle.distributed.rpc as rpc
>>> import os

>>> os.environ["PADDLE_WORKER_ENDPOINT"] = "127.0.0.1:9002"
>>> rpc.init_rpc("worker0", rank=0, world_size=1,
...             master_endpoint="127.0.0.1:8005")

>>> print(rpc.get_worker_info("worker0"))
{name: worker0, rank: 0, ip: 127.0.0.1, port: 9002}

>>> rpc.shutdown()

         