get_rank

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

返回当前进程的rank。

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

返回

(int) 当前进程的rank。

代码示例

  1. import paddle
  2. import paddle.distributed as dist
  3. # execute this command in terminal: export PADDLE_TRAINER_ID=0
  4. print("The rank is %d" % dist.get_rank())
  5. # The rank is 0