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)
|
gpu_name = torch.cuda.get_device_name(i)
|
||||||
if (
|
if (
|
||||||
"10" in gpu_name
|
"10" in gpu_name
|
||||||
|
or "16" in gpu_name
|
||||||
or "20" in gpu_name
|
or "20" in gpu_name
|
||||||
or "30" in gpu_name
|
or "30" in gpu_name
|
||||||
or "40" in gpu_name
|
or "40" in gpu_name
|
||||||
|
|
Loading…
Reference in New Issue