Detener grabación del historial de memoria CUDA(EndRecordCUDAMemoryHistory)
Este nodo técnicamente invoca la función torch de PyTorch.
End Record CUDA Memory History
El nodo End Recording CUDAMemory History se utiliza para detener la grabación y guardar una instantánea del historial de memoria CUDA. Normalmente se usa junto con un nodo que inicia la grabación, capturando los eventos de asignación y liberación de memoria GPU que ocurren entre ambos, y guardando los datos en un archivo. Este nodo está diseñado para ayudar a los usuarios a analizar y visualizar el uso de memoria GPU en flujos de trabajo de aprendizaje profundo.
Funcionalidad del Nodo
Técnicamente, este nodo genera el archivo de instantánea de memoria llamando a la función torch.cuda.memory._dump_snapshot() de PyTorch. Primero vacía la caché y luego genera un archivo .pt con marca de tiempo. Este formato de archivo es compatible con las herramientas oficiales de visualización de memoria de PyTorch y puede usarse para análisis posteriores en profundidad.
Descripción de Parámetros del Nodo - End Recording CUDAMemory History
Entradas de Conexión (Inputs)
| Nombre del Parámetro | Tipo de Dato | Requerido | Valor por Defecto | Rango/Opción | Descripción |
|---|---|---|---|---|---|
input | COMBO | Sí | - | - | Pasa la entrada recibida tal cual, manteniendo la conectividad del flujo de trabajo. |
Parámetros de Control (Parameters)
| Nombre del Parámetro | Tipo de Dato | Requerido | Valor por Defecto | Rango/Opción | Descripción |
|---|---|---|---|---|---|
output_path | CADENA | Sí | comfy_cuda_memory_history | - | La ruta de salida para el archivo completo de instantánea de memoria generado final. |
Salidas (Output)
| Nombre del Parámetro | Tipo de Dato | Descripción |
|---|---|---|
| input | ANY | Pasa la entrada recibida tal cual, manteniendo la conectividad del flujo de trabajo. |
| output_path | CADENA | La ruta de salida para el archivo completo de instantánea de memoria generado final. |
Casos de Uso
Al depurar flujos de trabajo complejos o modelos grandes, puedes colocar este nodo después de una secuencia de nodos que puedan consumir mucha memoria GPU. Por ejemplo, conectarlo al final de un flujo de trabajo que incluya múltiples pasos de generación y ampliación de imágenes de alta definición puede capturar las fluctuaciones de memoria durante todo el proceso de generación, ayudando a identificar fugas de memoria u optimizar el tamaño de los lotes.
Consideraciones
Es importante tener en cuenta que grabar instantáneas de memoria con frecuencia puede tener un ligero impacto en el rendimiento, por lo que se recomienda usar este nodo solo cuando sea necesario para depuración. El archivo de instantánea generado contiene el estado de la memoria en un momento específico y requiere herramientas especializadas (como PyTorch Memory Viz) o nodos de visualización complementarios para su interpretación.
Enlace al Código Fuente del Nodo End Recording CUDAMemory History
El nodo End Recording CUDAMemory History pertenece al paquete de nodos ComfyUI-KJNodes.
Comentarios
Inicia sesión con GitHub para unirte a la conversación.