Skip to content

Commit b951bdb

Browse files
authored
Update infer-web.py
1 parent 8d24582 commit b951bdb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

infer-web.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
if_gpu_ok = False
2727
for i in range(ngpu):
2828
gpu_name=torch.cuda.get_device_name(i)
29-
if("16"in gpu_name or "MX"in gpu_name):continue
29+
if(("16"in gpu_name and "V100"not in gpu_name) or "MX"in gpu_name):continue
3030
if("10"in gpu_name or "20"in gpu_name or "30"in gpu_name or "40"in gpu_name or "A50"in gpu_name.upper() 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 "TITAN"in gpu_name.upper()):#A10#A100#V100#A40#P40#M40#K80
3131
if_gpu_ok=True#至少有一张能用的N卡
3232
gpu_infos.append("%s\t%s"%(i,gpu_name))

0 commit comments

Comments
 (0)