Function monitor_gpu_memory_example

src/utils/gpu_monitor.py:218–247

使用示例

()

# 使用示例函数
def monitor_gpu_memory_example():
    """使用示例"""
    print("GPU显存监控示例")
    print("=" * 50)
    
    try:
        # 创建监控器
        monitor = GPUMemoryMonitor(
            gpu_index=0,      # 监控第一个GPU
            interval=0.5,     # 每0.5秒检查一次
            unit='GB'         # 使用GB作为单位
        )
        
        # 启动监控
        if monitor.start():
            # 这里可以运行你的GPU任务
            print("\n现在开始运行你的GPU任务...")
            print("监控正在后台进行")
            
            # 模拟一些工作（在实际使用中，这里应该是你的GPU任务）
            print("按回车键停止监控并获取峰值显存用量...")
            input()  # 等待用户按回车
            
            # 停止监控并获取结果
            peak_memory_gb = monitor.stop()
            
            print(f"\n监控完成！峰值显存用量: {peak_memory_gb:.3f} GB")
    
    except Exception as e:
        print(f"错误: {e}")


# 多GPU监控示例