- paddle.distributed. get_rank ( group=None ) [source]
Returns the rank of current trainer in the given group, ranks are consecutive integers in [0,
world_size). If none of the group is given, the global group will be used as default.
group (Group, optional) – The communication group you want to get rank of current trainer, use global group as default if group is None.
(int) The rank of current trainer in the given group. Return -1 if the process is not part of the given group.
grouponly supports in dygraph mode.
# Execute this script using distributed launch with one card configs. import paddle import paddle.distributed as dist dist.init_parallel_env() print("The rank is %d" % dist.get_rank()) # The rank is 0