diff --git a/slurm_gpustat/slurm_gpustat.py b/slurm_gpustat/slurm_gpustat.py index f3566a092313bb0d782cee65a57b023bf0758c91..7d1025727ae5739de1090dddaa15226e8306190f 100644 --- a/slurm_gpustat/slurm_gpustat.py +++ b/slurm_gpustat/slurm_gpustat.py @@ -547,7 +547,7 @@ def gpu_usage_grouped_by_user(resources: dict, partition: Optional[str] = None) if slurm_version.startswith("17"): resource_flag = "gres" else: - resource_flag = "tres-per-node" + resource_flag = "tres-per-job" cmd = f"squeue -O {resource_flag}:100,nodelist:100,username:100,jobid:100 --noheader" if partition: cmd += f" --partition={partition}"