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

12
gpuutil/gpuutil.py
View File

@ -127,12 +127,12 @@ def short_gpu_info(stat, disp_type='brief'):
util=stat_disp['util'], util=stat_disp['util'],
mem=stat_disp['mem'] mem=stat_disp['mem']
) )
if len(process_info) > 0: if len(process_info) > 0:
info += ' ' info += ' '
info += process_info info += process_info
return info return info
def get_basic_process_info_linux(): def get_basic_process_info_linux():
pipe = os.popen('ps axo user:20,pid,args:1024') pipe = os.popen('ps axo user:20,pid,args:1024')
@ -210,7 +210,7 @@ def draw_table(table, header_line = 0, c_align = 'r', h_align='c', delemeter = '
info += hline info += hline
info += hline info += hline
return info return info
class GPUStat(): class GPUStat():
@ -317,14 +317,14 @@ 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:
this_proc_str = proc_fmt.format( this_proc_str = proc_fmt.format(
user = procs[pid]['user'], user = procs[pid]['user'],
vmem = procs[pid]['vmem'].split(' ')[0], vmem = procs[pid]['vmem'].split(' ')[0],
pid = procs[pid]['pid'].rjust(5), pid = procs[pid]['pid'].rjust(5),
cmd = procs[pid]['command'], cmd = procs[pid]['command'],
gpus = ','.join(procs[pid]['gpu']) gpus = ','.join(procs[pid]['gpu'])
) )
@ -409,6 +409,6 @@ def auto_set(num, allow_nonfree=True, ask=True, blacklist=[], show=True):
else: else:
raise MoreGPUNeededError raise MoreGPUNeededError
set_gpu(selected_gpu, show=show) set_gpu(selected_gpu, show=show)
if __name__ == '__main__': if __name__ == '__main__':
print(get_basic_process_info_windows()) print(get_basic_process_info_windows())