End Recording CUDAMemory History(EndRecordCUDAMemoryHistory)
このノードは技術的にPyTorchのtorch。
End Record CUDA Memory History
End Recording CUDAMemory History ノードは、CUDAメモリ履歴の記録を停止し、スナップショットを保存するために使用されます。通常、記録を開始するノードとペアで使用され、両者の間で発生したGPUメモリの割り当てと解放イベントをキャプチャし、データをファイルに保存します。このノードは、ディープラーニングワークフローにおけるGPUメモリ使用状況の分析と可視化を支援することを目的としています。
ノード機能
技術的には、このノードはPyTorchの torch.cuda.memory._dump_snapshot() 関数を呼び出すことでメモリスナップショットファイルを生成します。最初にキャッシュをクリアし、タイムスタンプ付きの .pt ファイルを生成します。このファイル形式はPyTorch公式のメモリ可視化ツールと互換性があり、詳細な分析に使用できます。
ノードパラメータ説明 - End Recording CUDAMemory History
入力接続 (Inputs)
| パラメータ名 | データ型 | 必須 | デフォルト値 | 範囲/オプション | 説明 |
|---|---|---|---|---|---|
input | コンボ | はい | - | - | 受信した入力データをそのまま通過させ、ワークフローの接続性を維持します。 |
制御パラメータ (Parameters)
| パラメータ名 | データ型 | 必須 | デフォルト値 | 範囲/オプション | 説明 |
|---|---|---|---|---|---|
output_path | 文字列 | はい | comfy_cuda_memory_history | - | 最終的に生成された完全なメモリスナップショットファイルのパス。 |
出力
| パラメータ名 | データ型 | 説明 |
|---|---|---|
| input | ANY | 受信した入力データをそのまま通過させ、ワークフローの接続性を維持します。 |
| output_path | 文字列 | 最終的に生成された完全なメモリスナップショットファイルのパス。 |
使用シナリオ
複雑なワークフローや大規模モデルのデバッグ時に、大量のGPUメモリを消費する可能性のあるノードシーケンスの後にこのノードを配置できます。例えば、複数の高解像度画像生成とアップスケーリングステップを含むワークフローの末尾にこのノードを接続することで、生成プロセス全体におけるメモリの変動をキャプチャし、メモリリークの特定やバッチサイズの最適化に役立てることができます。
注意事項
メモリスナップショットを頻繁に記録すると、パフォーマンスにわずかな影響を与える可能性があるため、デバッグが必要な場合にのみこのノードを使用することを推奨します。生成されたスナップショットファイルには特定の時点のメモリ状態が含まれており、解釈には専用ツール（PyTorch Memory Vizなど）または対応する可視化ノードが必要です。
End Recording CUDAMemory History ノードソースコードリンク
End Recording CUDAMemory History ノードは ComfyUI-KJNodes ノードパッケージに含まれています。
コメント
GitHubでサインインしてディスカッションに参加しましょう。