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