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()