ComfyUI Memory Visualization：实时显存监控与热力图面板

ComfyUI Memory Visualization 是一款专为 ComfyUI 设计的实用工具节点，提供实时的 VRAM（显存）和系统内存监控。对于经常处理高分辨率图像、视频生成或大型模型的工作流开发者来说，它是排查“显存溢出（OOM）”错误和优化工作流性能的得力助手。

• GitHub：https://github.com/kijai/ComfyUI-MemoryVisualization

核心功能

1. 实时内存监控面板

• VRAM 使用率：实时显示 GPU 显存的已用、空闲和总量。
• 系统 RAM 使用率：监控主机内存使用情况，防止因系统内存不足导致的交换（Swap）卡顿。
• 直观展示：以清晰的面板形式嵌入 ComfyUI 界面，无需切换到任务管理器或终端日志。

2. 页面级驻留热图 (Page-level Residency Heatmap)

• 依赖 aimdo：当启用高级内存管理工具 aimdo 时，该插件能进一步显示每个模型在显存中的页面级驻留状态。
• 可视化热点：通过热力图直观展示哪些模型层或数据块当前正驻留在显存中，帮助用户理解模型的加载、卸载和缓存机制。
• 优化依据：识别显存碎片或未被正确释放的内存块，为调整 --lowvram 或 --medvram 参数提供数据支持。

3. 水印控制

• 提供简单的水印显示/隐藏控制，方便在截图或录屏分享工作流时保持界面整洁。

安装方法

只需将仓库克隆到 ComfyUI 的 custom_nodes 目录即可：

# 进入 ComfyUI 的 custom_nodes 文件夹
cd ComfyUI/custom_nodes

# 克隆仓库
git clone https://github.com/kijai/ComfyUI-MemoryVisualization.git

安装完成后，重启 ComfyUI，你将在界面中看到新增的内存监控面板。

为什么你需要它？

1. 告别盲目猜测：不再需要猜测是哪个节点导致了显存爆炸，实时监控让你一眼看出内存飙升的时刻。
2. 工作流优化：通过观察显存波动，你可以判断是否需要添加 Free VRAM 节点，或调整模型的加载顺序。
3. 调试利器：对于开发自定义节点的开发者，它能帮助验证内存是否正确释放，避免内存泄漏。
4. 配合 aimdo 进阶使用：如果你正在使用 aimdo 进行更细粒度的内存管理，此插件提供的热力图将是不可或缺的可视化工具。