From b3d934fe3a3e0053b325c76e3272390d8765abc7 Mon Sep 17 00:00:00 2001
From: Yury <rip95_95@mail.ru>
Date: Sat, 11 Jan 2025 20:14:49 +0200
Subject: [PATCH] Use NCCL and disable libuv for compatibility with latest
 pytorch versions

---
 rvc/train/train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rvc/train/train.py b/rvc/train/train.py
index 214ac065..0e19a6fc 100644
--- 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,