wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix GPU init in tests

Browse Source
This commit is contained in:
Eren Gölge
2021-09-10 08:28:10 +00:00
parent 0541a25e90
commit 3abc3a1d32
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/__init__.py
+2 -2
View File
@@ -7,8 +7,8 @@ from TTS.utils.generic_utils import get_cuda
def get_device_id():
use_cuda, _ = get_cuda()
if use_cuda:
if 'CUDA_VISIBLE_DEVICES' in os.environ and os.environ['CUDA_VISIBLE_DEVICES'] != "":
GPU_ID = os.environ['CUDA_VISIBLE_DEVICES'].split(',')[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: