Use NCCL and disable libuv for compatibility with latest pytorch vers…

…ions
IAHispano · Jan 11, 2025 · b3d934f · b3d934f
1 parent 9f7b7c1
commit b3d934f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -1,5 +1,6 @@
 import os
 import sys
+os.environ["USE_LIBUV"] = "0" if sys.platform == 'win32' else "1"
 import glob
 import json
 import torch
@@ -326,7 +327,7 @@ def run(
         writer_eval = None
 
     dist.init_process_group(
-        backend="gloo",
+        backend='gloo' if sys.platform == 'win32' or device.type != 'cuda' else 'nccl',
         init_method="env://",
         world_size=n_gpus if device.type == "cuda" else 1,
         rank=rank if device.type == "cuda" else 0,