Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2024-11-24 07:30:16 +01:00
Code Releases Wiki Activity

Update infer-web.py

Browse Source
This commit is contained in:
RVC-Boss 2023-04-26 15:39:19 +00:00 committed by GitHub
parent dc0c8756b5
commit 2ac8d553ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
infer-web.py
View File

@ -29,8 +29,6 @@ else:
if_gpu_ok = False
for i in range(ngpu):
gpu_name = torch.cuda.get_device_name(i)
if ("16" in gpu_name and "V100" not in gpu_name) or "MX" in gpu_name:
continue
if (
"10" in gpu_name
or "20" in gpu_name
@ -44,10 +42,10 @@ else:
or "70" in gpu_name
or "80" in gpu_name
or "90" in gpu_name
or "M4" in gpu_name
or "T4" in gpu_name
or "M4" in gpu_name.upper()
or "T4" in gpu_name.upper()
or "TITAN" in gpu_name.upper()
): # A10#A100#V100#A40#P40#M40#K80
): # A10#A100#V100#A40#P40#M40#K80#A4500
if_gpu_ok = True # 至少有一张能用的N卡
gpu_infos.append("%s\t%s" % (i, gpu_name))
gpu_info = (