We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dc0c875 commit 2ac8d55Copy full SHA for 2ac8d55
infer-web.py
@@ -29,8 +29,6 @@
29
if_gpu_ok = False
30
for i in range(ngpu):
31
gpu_name = torch.cuda.get_device_name(i)
32
- if ("16" in gpu_name and "V100" not in gpu_name) or "MX" in gpu_name:
33
- continue
34
if (
35
"10" in gpu_name
36
or "20" in gpu_name
@@ -44,10 +42,10 @@
44
42
or "70" in gpu_name
45
43
or "80" in gpu_name
46
or "90" in gpu_name
47
- or "M4" in gpu_name
48
- or "T4" in gpu_name
+ or "M4" in gpu_name.upper()
+ or "T4" in gpu_name.upper()
49
or "TITAN" in gpu_name.upper()
50
- ): # A10#A100#V100#A40#P40#M40#K80
+ ): # A10#A100#V100#A40#P40#M40#K80#A4500
51
if_gpu_ok = True # 至少有一张能用的N卡
52
gpu_infos.append("%s\t%s" % (i, gpu_name))
53
gpu_info = (
0 commit comments