Visualize CUDAMemory History(VisualizeCUDAMemoryHistory)
Este nodo depende técnicamente de `_memory_viz` de PyTorch.
Visualize CUDA Memory History
El nodo Visualize CUDAMemory History, que técnicamente depende de la función _memory_viz de PyTorch
El nodo Visualize CUDAMemory History se utiliza para convertir archivos de instantáneas del historial de asignación de memoria CUDA en gráficos interactivos visuales. Su función principal es leer un archivo pickle que contiene registros de uso de memoria y generar un informe HTML que se puede abrir en un navegador, ayudando al usuario a analizar visualmente el uso de la memoria GPU.
Funcionalidad del Nodo
Este nodo depende técnicamente de la función _memory_viz.trace_plot de PyTorch para procesar los datos de la instantánea. Acepta como entrada la ruta de un archivo de instantánea del historial de memoria CUDA guardado mediante serialización pickle. Internamente, analiza este archivo y genera contenido HTML que incluye gráficos de línea de tiempo. El archivo HTML generado se guarda automáticamente en la subcarpeta memory_history del directorio de salida de ComfyUI.
Explicación de los Parámetros del Nodo - Visualize CUDAMemory History
Parámetros de Control (Parameters)
| Nombre del Parámetro | Tipo de Dato | Obligatorio | Valor por Defecto | Rango de Valores/Opciones | Descripción |
|---|---|---|---|---|---|
snapshot_path | STRING | Sí | - | - | Especifica la ruta completa del archivo pickle que contiene los datos del historial de memoria CUDA. Este archivo suele ser generado por las herramientas de perfilado de memoria de PyTorch. |
Salida (Output)
| Nombre del Parámetro | Tipo de Dato | Descripción |
|---|---|---|
| output_path | STRING | Devuelve la dirección URL para acceder al informe de visualización HTML generado en el navegador. |
Casos de Uso
Al depurar u optimizar flujos de trabajo complejos, si se encuentran problemas de memoria GPU insuficiente o fugas de memoria, el usuario puede primero utilizar otras herramientas (como PyTorch Profiler) para capturar una instantánea de memoria y guardarla en un archivo. Luego, proporciona la ruta de este archivo de instantánea a este nodo. El nodo generará un informe visualizable, y el usuario puede abrir el enlace del informe en el navegador para ver en detalle la asignación y liberación de memoria en diferentes puntos en el tiempo, identificando así la causa raíz del problema.
Consideraciones
Tenga en cuenta que este nodo requiere como entrada una ruta válida a un archivo pickle de instantánea de memoria CUDA generado por una herramienta compatible. Si el formato del archivo es incorrecto o la ruta no es válida, el nodo podría no funcionar correctamente o generar un error. El enlace al informe HTML generado solo es válido mientras el servidor local de ComfyUI esté en ejecución.
Enlace al Código Fuente del Nodo Visualize CUDAMemory History
El nodo Visualize CUDAMemory History proviene del paquete de nodos ComfyUI-KJNodes.
Comentarios
Inicia sesión con GitHub para unirte a la conversación.