def METHOD_NAME(local_rank: int) -> None:
    # set the correct cuda visible devices (using pci order)
    os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
    all_gpu_ids = ",".join(str(x) for x in range(num_cuda_devices()))
    devices = os.getenv("CUDA_VISIBLE_DEVICES", all_gpu_ids)
    _log.info(f"LOCAL_RANK: {local_rank} - CUDA_VISIBLE_DEVICES: [{devices}]")