From 343aa6fc66604f0283ab89b29608573d1d86ff17 Mon Sep 17 00:00:00 2001 From: liujing04 <129054828+liujing04@users.noreply.github.com> Date: Sun, 16 Apr 2023 09:32:32 +0000 Subject: [PATCH] Update infer-web.py --- infer-web.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/infer-web.py b/infer-web.py index 54a1c4d..488dd7f 100644 --- a/infer-web.py +++ b/infer-web.py @@ -36,6 +36,10 @@ else: or "20" in gpu_name or "30" in gpu_name or "40" in gpu_name + or "A2" in gpu_name.upper() + or "A3" in gpu_name.upper() + or "A4" in gpu_name.upper() + or "P4" in gpu_name.upper() or "A50" in gpu_name.upper() or "70" in gpu_name or "80" in gpu_name