Fix get_gpu_name returning empty name. (pytorch#5925)

Repro in python shell >>> subprocess.check_output(["nvidia-smi", "--query-gpu=gpu_name", "--format=csv"], encoding='utf-8').split("\n")[1:] ['NVIDIA A100-SXM4-40GB', ''] Tested fix >>> subprocess.check_output(["nvidia-smi", "--query-gpu=gpu_name", "--format=csv"], encoding='utf-8').split("\n")[1:-1] ['NVIDIA A100-SXM4-40GB']
ManfeiBai · Dec 1, 2023 · 63b23e2 · 63b23e2
1 parent 6fcf15c
commit 63b23e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/util.py b/benchmarks/util.py
@@ -143,7 +143,7 @@ def get_cpu_name():
 def get_gpu_name():
   gpu_names = subprocess.check_output(
       ["nvidia-smi", "--query-gpu=gpu_name", "--format=csv"],
-      encoding='utf-8').split("\n")[1:]
+      encoding='utf-8').split("\n")[1:-1]
   if len(gpu_names) == 1:
     return gpu_names[0]
   return "One of " + ", ".join(gpu_names)