From a1fe2267af2df11cdaf28678af03fc958dc94a86 Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Thu, 2 Jan 2025 11:16:32 +0800 Subject: [PATCH] Update webui.py NVIDIA RTX 6000 Ada Generation --- webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.py b/webui.py index d510370..60d585e 100644 --- a/webui.py +++ b/webui.py @@ -72,7 +72,7 @@ mem = [] if_gpu_ok = False # 判断是否有能用来训练和加速推理的N卡 -ok_gpu_keywords={"10","16","20","30","40","A2","A3","A4","P4","A50","500","A60","70","80","90","M4","T4","TITAN","L4","4060","H"} +ok_gpu_keywords={"10","16","20","30","40","A2","A3","A4","P4","A50","500","A60","70","80","90","M4","T4","TITAN","L4","4060","H","600"} set_gpu_numbers=set() if torch.cuda.is_available() or ngpu != 0: for i in range(ngpu):