megengine.distributed.backend¶

backend = <property object>¶

用于获取或设置集合通信的后端。可选的后端包括：’nccl’, ‘shm’, ‘rccl’。

实际案例

import megengine.distributed as dist
dist.backend = "nccl"

megengine.distributed

组(Group)