get_world_size¶

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

返回指定通信组中的进程数，如果没有指定通信组，则默认使用全局通信组。

参数¶

group (Group，可选) - 指定想在得到哪个通信组下的进程数，如果没有指定，默认使用全局通信组。

返回¶

(int) 返回指定通信组中的进程数，如果当前进程不在该通信组中，则返回-1。

警告

参数 group 仅在动态图模式下支持

代码示例¶

          >>> # Execute this script using distributed launch with one card configs.
>>> import paddle
>>> import paddle.distributed as dist

>>> dist.init_parallel_env()
>>> print("The world_size is %d" % dist.get_world_size())
The world_size is 1

         