57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import psutil
|
||
from .logger import logger
|
||
|
||
# 尝试导入GPUtil，如果失败则设置has_gpu为False
|
||
try:
|
||
import GPUtil
|
||
GPUtil_available = True
|
||
except ImportError:
|
||
logger.warning("GPUtil模块未找到，将禁用GPU监控")
|
||
GPUtil_available = False
|
||
|
||
class SystemMonitor:
|
||
def __init__(self):
|
||
self.cpu_usage = 0.0
|
||
self.memory_usage = 0.0
|
||
self.gpu_memory_usage = 0.0
|
||
self.has_gpu = self._check_gpu()
|
||
|
||
def _check_gpu(self):
|
||
"""检查是否有可用的GPU"""
|
||
if not GPUtil_available:
|
||
return False
|
||
try:
|
||
gpus = GPUtil.getGPUs()
|
||
return len(gpus) > 0
|
||
except Exception as e:
|
||
logger.warning(f"GPU检查失败: {e}")
|
||
return False
|
||
|
||
def update_metrics(self):
|
||
"""更新系统资源使用情况"""
|
||
try:
|
||
# 更新CPU使用率
|
||
self.cpu_usage = psutil.cpu_percent(interval=0.1)
|
||
|
||
# 更新内存使用率
|
||
memory = psutil.virtual_memory()
|
||
self.memory_usage = memory.percent
|
||
|
||
# 更新GPU内存使用率（如果有GPU）
|
||
if self.has_gpu:
|
||
try:
|
||
gpus = GPUtil.getGPUs()
|
||
if gpus:
|
||
self.gpu_memory_usage = gpus[0].memoryUtil * 100
|
||
except Exception as e:
|
||
logger.warning(f"GPU内存使用率获取失败: {e}")
|
||
except Exception as e:
|
||
logger.error(f"系统资源监控更新失败: {e}")
|
||
|
||
def get_status_text(self):
|
||
"""获取状态栏显示文本"""
|
||
self.update_metrics()
|
||
status = f"CPU: {self.cpu_usage:.1f}% | 内存: {self.memory_usage:.1f}%"
|
||
if self.has_gpu:
|
||
status += f" | GPU内存: {self.gpu_memory_usage:.1f}%"
|
||
return status |