get_current_worker_info

paddle.distributed.rpc. get_current_worker_info ( ) [源代码]

获取当前 worker 的信息。

参数

返回

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

代码示例

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

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

>>> print(rpc.get_current_worker_info())
{name: worker0, rank: 0, ip: 127.0.0.1, port: 9004}

>>> rpc.shutdown()