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):