import os import torch.distributed as dist dist.init_process_group("gloo")