def ddp_setup()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

def ddp_setup():
    init_process_group(backend="nccl")
    torch.cuda.set_device(int(os.environ["LOCAL_RANK"]))