Torch configuration for Windows

[sevaraMS]

How can I adopt this below part for Windows?. As I know this part for Ubuntu
def train():
args = parse_args()
torch.cuda.set_device(args.local_rank)
dist.init_process_group(
backend = 'nccl',
init_method = 'tcp://127.0.0.1:33241',
world_size = torch.cuda.device_count(),
rank=args.local_rank
)
setup_logger(respth)