paddle.distributed / get_world_size

get_world_size¶

paddle.distributed. get_world_size ( ) [源代码] ¶

返回参与当前任务的进程数。

当前进程数等于环境变量 PADDLE_TRAINERS_NUM 的值，默认值为1。

返回¶

(int) 参与任务的进程数。

代码示例¶

import paddle
import paddle.distributed as dist

# execute this command in terminal: export PADDLE_TRAINERS_NUM=4
print("The world_size is %d" % dist.get_world_size())
# The world_size is 4

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle.distributed / get_world_size

get_world_size¶

返回¶

代码示例¶