From 2ac8d553ab5e330acc3bd6167e0de72ba41b9f15 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Wed, 26 Apr 2023 15:39:19 +0000
Subject: [PATCH] Update infer-web.py

---
 infer-web.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/infer-web.py b/infer-web.py
index 040cb7f..f85ed05 100644
--- a/infer-web.py
+++ b/infer-web.py
@@ -29,8 +29,6 @@ else:
     if_gpu_ok = False
     for i in range(ngpu):
         gpu_name = torch.cuda.get_device_name(i)
-        if ("16" in gpu_name and "V100" not in gpu_name) or "MX" in gpu_name:
-            continue
         if (
             "10" in gpu_name
             or "20" in gpu_name
@@ -44,10 +42,10 @@ else:
             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 "M4" in gpu_name.upper()
+            or "T4" in gpu_name.upper()
             or "TITAN" in gpu_name.upper()
-        ):  # A10#A100#V100#A40#P40#M40#K80
+        ):  # A10#A100#V100#A40#P40#M40#K80#A4500
             if_gpu_ok = True  # 至少有一张能用的N卡
             gpu_infos.append("%s\t%s" % (i, gpu_name))
 gpu_info = (