fix
This commit is contained in:
parent
9d64eaf051
commit
b0d7618592
|
@ -296,7 +296,7 @@ class Specs:
|
||||||
gpu_str, gpu_mem, gpus, nvml_err = get_gpu_info()
|
gpu_str, gpu_mem, gpus, nvml_err = get_gpu_info()
|
||||||
if require_same_gpus:
|
if require_same_gpus:
|
||||||
last_gpu_name=''
|
last_gpu_name=''
|
||||||
for gpu in gpus:
|
for gpu in gpus["nvidia"]:
|
||||||
if not last_gpu_name:
|
if not last_gpu_name:
|
||||||
last_gpu_name=gpu["name"]
|
last_gpu_name=gpu["name"]
|
||||||
elif last_gpu_name!=gpu["name"]:
|
elif last_gpu_name!=gpu["name"]:
|
||||||
|
|
Loading…
Reference in New Issue