Use pyworld for pitch

2026-09-09 11:16:00 +08:00 · 2021-09-06 15:16:58 +00:00
parent c1513ec4cd
commit 2c4bbbf9b9
6 changed files with 19 additions and 163 deletions
@@ -7,7 +7,10 @@ from TTS.utils.generic_utils import get_cuda
 def get_device_id():
    use_cuda, _ = get_cuda()
    if use_cuda:
-        GPU_ID = "0"
+        if 'CUDA_VISIBLE_DEVICES' in os.environ and os.environ['CUDA_VISIBLE_DEVICES'] != "":
+            GPU_ID = os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0]
+        else:
+            GPU_ID = "0"
    else:
        GPU_ID = ""
    return GPU_ID