zmyme/gpuutil
1
0
Fork 0
Code Issues Pull Requests Releases 5 Wiki Activity

fix error when multi process on a single gpu

Browse Source
This commit is contained in:
mingyang 2020-11-19 10:22:51 +08:00
parent 9e3bae50ba
commit 356d201499
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gpuutil/gpuutil.py
View File

@ -317,7 +317,7 @@ class GPUStat():
if pid not in procs: if pid not in procs:
procs[pid] = proc procs[pid] = proc
else: else:
procs[pid].append(gpu['id']) procs[pid]['gpu'].append(gpu['id'])
proc_fmt = '[{pid}|{gpus}] {user}({vmem} MiB) {cmd}' proc_fmt = '[{pid}|{gpus}] {user}({vmem} MiB) {cmd}'
proc_strs = [] proc_strs = []
for pid in procs: for pid in procs: