get_rank

paddle.distributed. get_rank ( ) [源代码]

返回当前进程的 rank。

当前进程 rank 的值等于环境变量 PADDLE_TRAINER_ID 的值,默认值为 0。

返回

(int) 当前进程的 rank。

代码示例

# 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