Update infer-web.py
This commit is contained in:
parent
9068d5283e
commit
211a842e88
|
@ -34,6 +34,7 @@ else:
|
|||
gpu_name = torch.cuda.get_device_name(i)
|
||||
if (
|
||||
"10" in gpu_name
|
||||
or "16" in gpu_name
|
||||
or "20" in gpu_name
|
||||
or "30" in gpu_name
|
||||
or "40" in gpu_name
|
||||
|
|
Loading…
Reference in New Issue