This commit is contained in:
allegroai 2020-09-09 22:07:50 +03:00
parent 24ab1ecc2d
commit 33f42def54

View File

@ -15,10 +15,8 @@ from __future__ import print_function
from __future__ import unicode_literals from __future__ import unicode_literals
import json import json
import os.path
import platform import platform
import sys import sys
import time
from datetime import datetime from datetime import datetime
import psutil import psutil
@ -199,10 +197,10 @@ class GPUStatCollection(object):
if nv_process.pid not in GPUStatCollection.global_processes: if nv_process.pid not in GPUStatCollection.global_processes:
GPUStatCollection.global_processes[nv_process.pid] = \ GPUStatCollection.global_processes[nv_process.pid] = \
psutil.Process(pid=nv_process.pid) psutil.Process(pid=nv_process.pid)
ps_process = GPUStatCollection.global_processes[nv_process.pid]
process['pid'] = nv_process.pid process['pid'] = nv_process.pid
# noinspection PyBroadException # noinspection PyBroadException
try: try:
# ps_process = GPUStatCollection.global_processes[nv_process.pid]
# we do not actually use these, so no point in collecting them # we do not actually use these, so no point in collecting them
# process['username'] = ps_process.username() # process['username'] = ps_process.username()
# # cmdline returns full path; # # cmdline returns full path;