fix: The GPU model is written incorrectly, and the calculation result…

… is misjudged.
arthur1573 · Mar 18, 2024 · 9034711 · 9034711
1 parent 9dbd93d
commit 9034711
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/scripts/gpu_capabilities.json b/scripts/gpu_capabilities.json
@@ -54,8 +54,8 @@
     "NVIDIA A10": 8.6,
     "NVIDIA A16": 8.6,
     "NVIDIA A2": 8.6,
-    "NVIDIA T4": 7.5,
-    "NVIDIA V100": 7.0,
+    "Tesla T4": 7.5,
+    "Tesla V100": 7.0,
     "Tesla P100": 6.0,
     "Tesla P40": 6.1,
     "Tesla P4": 6.1

diff --git a/scripts/nvidia_gpus_compute_capability.py b/scripts/nvidia_gpus_compute_capability.py
@@ -56,8 +56,8 @@
     ("NVIDIA A10", 8.6),
     ("NVIDIA A16", 8.6),
     ("NVIDIA A2", 8.6),
-    ("NVIDIA T4", 7.5),
-    ("NVIDIA V100", 7.0),
+    ("Tesla T4", 7.5),
+    ("Tesla V100", 7.0),
     ("Tesla P100", 6.0),
     ("Tesla P40", 6.1),
     ("Tesla P4", 6.1),

diff --git a/scripts/run_for_local_option.sh b/scripts/run_for_local_option.sh
@@ -154,7 +154,7 @@ if ! command -v jq &> /dev/null; then
 fi
 compute_capability=$(jq -r ".[\"$gpu_model\"]" /workspace/qanything_local/scripts/gpu_capabilities.json)
 # 如果compute_capability为空，则说明显卡型号不在gpu_capabilities.json中
-if [ -z "$compute_capability" ]; then
+if [ "$compute_capability" == "null" ]; then
     echo "您的显卡型号 $gpu_model 不在支持列表中，请联系技术支持。"
     exit 1
 fi