diff --git a/finetune_speaker_v2.py b/finetune_speaker_v2.py index 49ebdd7..85fa044 100644 --- a/finetune_speaker_v2.py +++ b/finetune_speaker_v2.py @@ -65,7 +65,8 @@ def run(rank, n_gpus, hps): writer = SummaryWriter(log_dir=hps.model_dir) writer_eval = SummaryWriter(log_dir=os.path.join(hps.model_dir, "eval")) - dist.init_process_group(backend='nccl', init_method='env://', world_size=n_gpus, rank=rank) + # Use gloo backend on Windows for Pytorch + dist.init_process_group(backend= 'gloo' if os.name == 'nt' else 'nccl', init_method='env://', world_size=n_gpus, rank=rank) torch.manual_seed(hps.train.seed) torch.cuda.set_device(rank) @@ -317,4 +318,4 @@ def evaluate(hps, generator, eval_loader, writer_eval): if __name__ == "__main__": - main() \ No newline at end of file + main() diff --git a/requirements.txt b/requirements.txt index 975d78a..bf44056 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ Cython -librosa +librosa==0.9.1 numpy scipy tensorboard