Visualize CUDAMemory History(VisualizeCUDAMemoryHistory)

Visualize CUDAMemory History ノードは、技術的に PyTorch の _memory_viz に依存しています。

Visualize CUDAMemory History ノードは、CUDA メモリ割り当て履歴のスナップショットファイルを、視覚的なインタラクティブなグラフに変換するために使用されます。その中核機能は、メモリ使用記録を含む pickle ファイルを読み取り、ブラウザで開くことができる HTML レポートを生成し、ユーザーが GPU メモリの使用状況を直感的に分析できるようにすることです。

ノード機能

このノードは、技術的に PyTorch の _memory_viz.trace_plot 関数に依存してスナップショットデータを処理します。pickle シリアライズで保存された CUDA メモリ履歴スナップショットファイルのパスを入力として受け取り、内部でこのファイルを解析し、タイムラインチャートを含む HTML コンテンツを生成します。生成された HTML ファイルは、ComfyUI の出力ディレクトリ内の memory_history サブフォルダに自動的に保存されます。

ノードパラメータ説明 - Visualize CUDAMemory History

コントロールパラメータ (Parameters)

出力

使用シナリオ

複雑なワークフローのデバッグや最適化中に、GPU メモリ不足やメモリリークの問題が発生した場合、ユーザーはまず他のツール（PyTorch Profiler など）を使用してメモリスナップショットをキャプチャし、ファイルとして保存できます。その後、このスナップショットファイルのパスをこのノードに提供すると、ノードはビジュアライゼーションレポートを生成します。ユーザーはブラウザでレポートリンクを開き、異なる時点でのメモリ割り当てと解放の詳細を確認することで、問題の根本原因を特定できます。

注意事項

このノードは、互換性のあるツールによって生成された有効な CUDA メモリスナップショット pickle ファイルのパスを入力する必要があります。ファイル形式が正しくない場合やパスが無効な場合、ノードは正常に動作しないか、エラーが発生する可能性があります。生成された HTML レポートリンクは、ローカルの ComfyUI サーバーが実行されている間のみ有効です。

Visualize CUDAMemory History ノードソースコードリンク

Visualize CUDAMemory History ノードは ComfyUI-KJNodes ノードパッケージに由来します。