公司对比电脑性能需要获取程序所占内存的提交大小
psutil库Windows获取的cpu实时速度是基准速度,只有Linux里是实时的,所以是无效参数
import psutil
def get_mem(pid):
p = psutil.Process(pid)
# cpu占用百分比
cpu_percent = p.cpu_percent() / psutil.cpu_count()
# 程序运行内存的提交大小, 单位K,与任务管理器一致
mem = p.memory_info().vms / 1024
# GPU信息
gpu = GPUtil.getGPUs()[0]
# 已用显存
gpu_used = gpu.memoryUsed
# 显存占用率
gpu_util = gpu.memoryUtil * 100
# GPU利用率
gpu_load = gpu.load * 100
return cpu_percent, mem, gpu_used, gpu_util, gpu_load