code: def setup(rank, world_size):
      os.environ["MASTER_ADDR"] = "localhost"
      os.environ["MASTER_PORT"] = "12355"
  
      # initialize the process group
      dist.init_process_group("gloo", rank=rank, world_size=world_size)
  
file_path: wandb/tests/functional_tests/t0_main/torch/t3_ddp_basic.py
imports: import os
  import torch
  import torch.distributed
  import torch.multiprocessing
  import torch.nn
  import torch.optim
  import wandb
  from torch.nn.parallel import DistributedDataParallel
label_desc: Use image node_id 1 for calling a global function with example usage: setup(rank, world_size) without return types