fix

2024-05-12 11:20:34 +00:00 · 2024-05-12 11:20:34 +00:00 · b0d7618592
parent 9d64eaf051
commit b0d7618592
1 changed files with 1 additions and 1 deletions
--- a/lib/get_specs.py
+++ b/lib/get_specs.py
@ -296,7 +296,7 @@ class Specs:
        gpu_str, gpu_mem, gpus, nvml_err = get_gpu_info()
        if require_same_gpus:
            last_gpu_name=''
-            for gpu in gpus:
+            for gpu in gpus["nvidia"]:
                if not last_gpu_name:
                    last_gpu_name=gpu["name"]
                elif last_gpu_name!=gpu["name"]: