Function quick_monitor

src/utils/gpu_monitor.py:310–339

快速启动监控的便捷函数 Args: gpu_index: GPU索引 interval: 监控间隔 unit: 单位 wait_for_input: 是否等待用户输入 Returns: float: 峰值显存使用量

(gpu_index=0, interval=0.5, unit='GB', wait_for_input=True)

# 快速使用函数
def quick_monitor(gpu_index=0, interval=0.5, unit='GB', wait_for_input=True):
    """
    快速启动监控的便捷函数
    
    Args:
        gpu_index: GPU索引
        interval: 监控间隔
        unit: 单位
        wait_for_input: 是否等待用户输入
    
    Returns:
        float: 峰值显存使用量
    """
    monitor = GPUMemoryMonitor(gpu_index=gpu_index, interval=interval, unit=unit)
    
    try:
        if monitor.start():
            if wait_for_input:
                print("按回车键停止监控...")
                input()
            else:
                # 如果不等待用户输入，这里可以设置其他停止条件
                # 例如：监控特定时间或直到某个条件满足
                print("监控已启动，将在后台运行")
                print("调用 monitor.stop() 来停止并获取结果")
                return monitor
        return None
    except Exception as e:
        print(f"监控失败: {e}")
        return None


# 测试代码