pmonitor 1.4.2__py3-none-any.whl → 1.4.3__py3-none-any.whl

monitor/mac_gpu.py

@@ -36,21 +36,26 @@ if platform.system() != 'Windows':
         try:
             gpu_info = accelerator_performance_statistics()
             use_gpu_memory = 0
-            device_usage = 0
-            # print(f"gpu_info: {gpu_info['PerformanceStatistics']}")
+            total_gpu_memory = 0
+            free_gpu_memory = 0
             for key, value in gpu_info["PerformanceStatistics"].items():
-                if key == "In use system memory":
-                    use_gpu_memory = int(value) / 1024 / 1024
-                    # print(f"{key}: {value}")
-                elif key == "inUseSysMemoryBytes":
-                    use_gpu_memory = int(value) / 1024 / 1024
-                    # print(f"{key}: {value}")
-                elif key == "Device Utilization %":
-                    device_usage = int(value)
+                if key == "In use system memory" or key == "gartUsedBytes":
+                    use_gpu_memory = int(value)
+                elif key == "gartSizeBytes":
+                    total_gpu_memory = int(value)
+                elif key == "gartFreeBytes":
+                    free_gpu_memory = int(value)
+                elif key == "Alloc system memory":
+                    total_gpu_memory = int(value)
                     # print('device usage', device_usage)
             # use_gpu_memory = gpu_info["PerformanceStatistics"]["In use system memory"]
             # total_gpu_memory = gpu_info["PerformanceStatistics"]["Alloc system memory"]
         except Exception as e:
             print(f"Mac端在arm64平台通过接口获取Gpu Memory失败：{e}")
             use_gpu_memory = 0
-        return use_gpu_memory, device_usage
+            total_gpu_memory = 0
+            free_gpu_memory = 0
+        if free_gpu_memory == 0:
+            free_gpu_memory = total_gpu_memory - use_gpu_memory
+        return round(use_gpu_memory / 1024 / 1024, 2), round(total_gpu_memory / 1024 / 1024, 2), round(
+            free_gpu_memory / 1024 / 1024, 2)

monitor/monitor_pids.py

@@ -7,9 +7,7 @@ import asyncio
 import platform
 import ctypes
 import os
-import stat
-import subprocess
-import re
+
 if platform.system() != 'Windows':
     from monitor.mac_gpu import get_gpu_memory
 
@@ -28,15 +26,18 @@ class PidsPerf:
             minor_real_mem_sum = 0
             minor_mem_percent_sum = 0
             minor_thread_count_sum = 0
-            gpu_memory_usage_0 = 0
-            gpu_memory_percent_0 = 0
-            gpu_memory_usage_1 = 0
-            gpu_memory_percent_1 = 0
+            gpu_memory_usage = 0
+            gpu_memory_total = 0
+            gpu_memory_free = 0
             current_pids = self.processUtil.get_children_pids(current_pid)  # 获取所有子进程
             current_pids.add(current_pid)  # 将主进程及子进程添加到list中
             pids_process = self.processUtil.get_pids_process(current_pids)  # 获取所有进程
 
-            gpu_memory_usage_0, gpu_memory_percent_0 = get_gpu_memory()
+            mem_info = psutil.virtual_memory()
+            mem_total = round(mem_info.total / 1024 / 1024, 2)  # 总虚拟内存
+            mem_available = round(mem_info.available / 1024 / 1024, 2)  # 剩余虚拟内存
+
+            gpu_memory_usage, gpu_memory_total, gpu_memory_free = get_gpu_memory()
             for process in pids_process:
                 try:
                     cpu_percent = process.cpu_percent()
@@ -77,9 +78,10 @@ class PidsPerf:
             io_read_bytes = io_read_bytes_end - io_read_bytes_start  # io read/byte
             io_write_bytes = io_write_bytes_end - io_write_bytes_start  # io write/byte
             disk_data = {'io read': io_read_bytes, 'io write': io_write_bytes}
-            gpu_data = {'gpu 0 memory usage': gpu_memory_usage_0, 'gpu 0 device Utilization': gpu_memory_percent_0,
-                        'gpu 1 memory usage': gpu_memory_usage_1, 'gpu 1 device Utilization': gpu_memory_percent_1}
-            data = {'main': main_data, 'other': minor_data, 'disk': disk_data, "gpu": gpu_data}
+            gpu_data = {'gpu memory usage': gpu_memory_usage, 'gpu memory total': gpu_memory_total,
+                        'gpu memory free': gpu_memory_free}
+            data = {'main': main_data, 'other': minor_data, 'disk': disk_data, "gpu": gpu_data,
+                    "memory_total": mem_total, "memory_available": mem_available}
             json_data = json.dumps(data)
             print(json_data)
             sys.stdout.flush()
@@ -101,6 +103,7 @@ class PidsPerf:
     def get_win_perf(self):
         from monitor.WinPidUtil import PidWinPerformance, FILETIME
         from ctypes import CDLL, c_int, c_char_p, byref
+        from PyLibreHardwareMonitor.computer import Computer
         pidWinPerf = PidWinPerformance()
         system_time_pre = pidWinPerf.filetime_to_100nano_seconds(FILETIME())
         p_pid = self.processUtil.find_main_process_pid(self.process_name)
@@ -114,6 +117,9 @@ class PidsPerf:
         gpu_monitor_dll = ctypes.CDLL(dll_path)
         gpu_monitor_dll.GetGPUDataForProcess.argtypes = [ctypes.c_ulong]
         gpu_monitor_dll.GetGPUDataForProcess.restype = ctypes.c_char_p
+        # 获取GPU物理内存
+        computer = Computer()
+        gpu_key = list(computer.gpu.keys())[0]
         while True:
             minor_cpu_sum = 0
             minor_workSet_mem_sum = 0
@@ -145,6 +151,7 @@ class PidsPerf:
             child_pids.add(p_pid)  # app内所有pid
             child_pids = {pid for pid in child_pids if psutil.pid_exists(pid)}  # 移除不存在的进程
 
+            small_data = computer.gpu[gpu_key]['SmallData']  # 获取GPU物理内存信息
             try:
                 new_pids_cpu_time_pre = {pid: pidWinPerf.get_process_cpu_time(pid) for pid in child_pids if
                                          pid not in process_cpu_times_pre}
@@ -156,6 +163,8 @@ class PidsPerf:
 
             # print('----------------------------------------------------------')
             # print('pid', child_pids)
+            memory_total = round(psutil.virtual_memory().total / 1024 / 1024, 2)
+            memory_available = round(psutil.virtual_memory().available / 1024 / 1024, 2)
 
             for pid in child_pids:
                 try:
@@ -241,8 +250,14 @@ class PidsPerf:
                           'gpu_System': round(minor_gpu_system_sum, 2),
                           'gpu_Committed': round(minor_gpu_committed_sum, 2),
                           'gpu_Usage': round(minor_gpu_percent_sum, 2)}
+            gpu_data = {'gpu memory usage': small_data['GPU Memory Used'],
+                        'gpu memory total': small_data['GPU Memory Total'],
+                        'gpu memory free': small_data['GPU Memory Free']}
             try:
-                data = {'main': main_data, 'other': other_data, 'disk': disk_data}
+                data = {'main': main_data, 'other': other_data, 'disk': disk_data,
+                        'gpu': gpu_data,
+                        'memory_total': memory_total,
+                        'memory_available': memory_available}
                 json_data = json.dumps(data)
                 print(json_data)
             except UnboundLocalError:

monitor/run_monitor.py

@@ -27,7 +27,7 @@ def main():
     if args.time:
         interval = args.time
 
-    # process = 'pixcake.exe'
+    # process = 'ColorByte.exe'
     # interval = 1
     process_name = ProcessName()
     if process is not None:

{pmonitor-1.4.2.dist-info → pmonitor-1.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pmonitor
-Version: 1.4.2
+Version: 1.4.3
 Summary: pc monitor
 Home-page: UNKNOWN
 Author: cfr

{pmonitor-1.4.2.dist-info → pmonitor-1.4.3.dist-info}/RECORD

@@ -1,11 +1,11 @@
 monitor/WinPidUtil.py,sha256=0vu2cqcQRrBTb5t_9HOmUoOfbe9KRrI-cyDlJX9bcrw,8753
 monitor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monitor/get_process_name.py,sha256=HxePo2gdrTo2Ukg4DTnYzUMdczVWywCeLCXQ-RRWyu8,3913
-monitor/mac_gpu.py,sha256=dM0KtLKos5ZIV0JV7vtNQMV3QvB4L05KBH1y9nryk-Q,2089
+monitor/mac_gpu.py,sha256=kQDPMW04lIYPBbJw772Jh1OQxj-D4JNmdjJlP6BuR4w,2320
 monitor/monitor_mac.py,sha256=OoptTTXTZQupzBwgGOfKNjyYAKBxDy-wI3l6T7XVE8s,4651
-monitor/monitor_pids.py,sha256=6Y75WJUHQgLbNEuKUfviu46JHDFANtHggqvdAJVykrg,14608
+monitor/monitor_pids.py,sha256=yvUOIuwMw5stDxOVSY2vu72CzxZzmL6qCYsF9vw3VY8,15580
 monitor/monitor_win.py,sha256=xs5nzqqEPoDmJTegh3lQhVjjpPcOWnruWKK65ttqnTo,6161
-monitor/run_monitor.py,sha256=JKjb0X2tjPECFg_k7oby7Yg5Xhs8UNmbjCBP_6oXqMk,1579
+monitor/run_monitor.py,sha256=s9JSFu5olh3VhrEqj_CZ5l25nvBtd2cN0NhU9R2KOKM,1581
 monitor/sys_info.py,sha256=aNultuRoQuRYPkYo397xAXVDXP07Qx5JOHtYzNmEvuc,3208
 monitor/DLL/GpuMonitorLib.dll,sha256=xbH4eyDvnv2g6go1fvUt6qQd7qA9n6piP8My4XubLPQ,66560
 monitor/DLL/KernelBase.dll,sha256=MSJ2WSh-2Lk6yFVLX_a8G2EnNuaOtJGpfg-Vw9HY_go,3860744
@@ -24,8 +24,8 @@ monitor/DLL/ucrtbased.dll,sha256=vtmKFPEHyr2OXkrUOu3Qs1dlbKG1dxZ8ItKCkTTU5S4,223
 monitor/DLL/vcruntime140.dll,sha256=AsaqDm5iRBGp8ZsDYKeGWrFZCOJgJFEOXDipwINiw1o,119888
 monitor/DLL/vcruntime140_1.dll,sha256=fdmqAuJxxoym1fGNZR0joV1yWXFa9DMmV4993ifzdjc,49640
 monitor/exec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pmonitor-1.4.2.dist-info/METADATA,sha256=01Bx2eDmkJ5cP9cILiaN0LuuleEKTg6JhoDYovXMbx4,215
-pmonitor-1.4.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pmonitor-1.4.2.dist-info/entry_points.txt,sha256=vmlLEANgf2fZar9BeXYiKdF6GMJbVXip-SIZx5yPXDo,55
-pmonitor-1.4.2.dist-info/top_level.txt,sha256=tGkQDkVeyKgP5Rr7acpp0df83NBAnI8up0sGwRxuuQ4,8
-pmonitor-1.4.2.dist-info/RECORD,,
+pmonitor-1.4.3.dist-info/METADATA,sha256=fqc-_49ex8Sv1XfkptmGPMP67m5BT0pt28YhTSV-fJI,215
+pmonitor-1.4.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pmonitor-1.4.3.dist-info/entry_points.txt,sha256=vmlLEANgf2fZar9BeXYiKdF6GMJbVXip-SIZx5yPXDo,55
+pmonitor-1.4.3.dist-info/top_level.txt,sha256=tGkQDkVeyKgP5Rr7acpp0df83NBAnI8up0sGwRxuuQ4,8
+pmonitor-1.4.3.dist-info/RECORD,,