Visualize CUDAMemory History (VisualizeCUDAMemoryHistory) - ComfyUI-KJNodes

Узел Visualize CUDAMemory History, который технически зависит от функции PyTorch `_memory_viz

Узел Visualize CUDAMemory History предназначен для преобразования файлов снимков истории выделения CUDA-памяти в интерактивные визуальные диаграммы. Его основная функция — чтение pickle-файла, содержащего записи об использовании памяти, и генерация HTML-отчета, который можно открыть в браузере, чтобы помочь пользователю интуитивно анализировать использование GPU-памяти.

Функциональность узла

Технически этот узел зависит от функции PyTorch _memory_viz.trace_plot для обработки данных снимков. Он принимает в качестве входных данных путь к файлу снимка истории CUDA-памяти, сохраненному с помощью сериализации pickle. Внутренне он анализирует этот файл и генерирует HTML-контент, содержащий диаграммы временной шкалы. Сгенерированный HTML-файл автоматически сохраняется в подпапке memory_history выходного каталога ComfyUI.

Описание параметров узла - Visualize CUDAMemory History

Параметры элементов управления (Parameters)

Имя параметра	Тип данных	Обязательный	Значение по умолчанию	Допустимые значения/опции	Описание
`snapshot_path`	STRING	Да	-	-	Указывает полный путь к pickle-файлу, содержащему данные истории CUDA-памяти. Этот файл обычно генерируется инструментами профилирования памяти PyTorch.

Выходные данные (Output)

Имя параметра	Тип данных	Описание
output_path	STRING	Возвращает URL-адрес для доступа к сгенерированному HTML-отчету с визуализацией в браузере.

Сценарии использования

При отладке или оптимизации сложных рабочих процессов, если возникают проблемы с нехваткой GPU-памяти или утечками памяти, пользователь может сначала использовать другие инструменты (например, PyTorch Profiler) для захвата снимков памяти и сохранения их в файл. Затем, передав путь к этому файлу снимка данному узлу, узел сгенерирует отчет с визуализацией. Пользователь может открыть ссылку на отчет в браузере, чтобы детально изучить выделение и освобождение памяти в различные моменты времени и определить коренную причину проблемы.

Важные замечания

Обратите внимание, что этот узел требует ввода действительного пути к pickle-файлу снимка CUDA-памяти, сгенерированному совместимым инструментом. Если формат файла неверен или путь недействителен, узел может работать некорректно или выдавать ошибку. Сгенерированная ссылка на HTML-отчет действительна только при запущенном локальном сервере ComfyUI.

Ссылка на исходный код узла Visualize CUDAMemory History

Узел Visualize CUDAMemory History взят из пакета узлов ComfyUI-KJNodes.

Visualize CUDAMemory History(VisualizeCUDAMemoryHistory)