Visualize CUDAMemory History(VisualizeCUDAMemoryHistory)
Ce nœud dépend techniquement de `_memory_viz` de PyTorch.
Visualize CUDA Memory History
Le nœud Visualize CUDAMemory History, qui dépend techniquement de la fonction _memory_viz de PyTorch.
Le nœud Visualize CUDAMemory History est utilisé pour convertir un fichier d'instantané de l'historique d'allocation de la mémoire CUDA en un graphique interactif et visuel. Sa fonction principale est de lire un fichier pickle contenant des enregistrements d'utilisation de la mémoire et de générer un rapport HTML qui peut être ouvert dans un navigateur, aidant ainsi l'utilisateur à analyser visuellement l'utilisation de la mémoire GPU.
Fonctionnalité du Nœud
Ce nœud dépend techniquement de la fonction _memory_viz.trace_plot de PyTorch pour traiter les données de l'instantané. Il accepte en entrée un chemin de fichier d'instantané de l'historique de la mémoire CUDA, sauvegardé via une sérialisation pickle. En interne, il analyse ce fichier et génère un contenu HTML contenant un graphique chronologique. Le fichier HTML généré est automatiquement enregistré dans le sous-dossier memory_history du répertoire de sortie de ComfyUI.
Explication des Paramètres du Nœud - Visualize CUDAMemory History
Paramètres des Contrôles (Parameters)
| Nom du Paramètre | Type de Données | Requis | Valeur par Défaut | Plage de Valeurs/Options | Description |
|---|---|---|---|---|---|
snapshot_path | STRING | Oui | - | - | Spécifie le chemin complet du fichier pickle contenant les données historiques de la mémoire CUDA. Ce fichier est généralement généré par les outils de profilage de mémoire de PyTorch. |
Sortie (Output)
| Nom du Paramètre | Type de Données | Description |
|---|---|---|
| output_path | STRING | Retourne l'URL d'accès au rapport de visualisation HTML généré, à ouvrir dans un navigateur. |
Scénarios d'Utilisation
Lors du débogage ou de l'optimisation de flux de travail complexes, si des problèmes de mémoire GPU insuffisante ou de fuite de mémoire surviennent, l'utilisateur peut d'abord utiliser d'autres outils (comme PyTorch Profiler) pour capturer un instantané de la mémoire et le sauvegarder dans un fichier. Ensuite, en fournissant le chemin de ce fichier d'instantané à ce nœud, celui-ci générera un rapport de visualisation. L'utilisateur peut ouvrir le lien du rapport dans son navigateur pour examiner en détail les allocations et libérations de mémoire à différents moments, afin d'identifier la source du problème.
Points d'Attention
Veuillez noter que ce nœud nécessite en entrée un chemin de fichier d'instantané de la mémoire CUDA valide, généré par un outil compatible. Si le format du fichier est incorrect ou si le chemin est invalide, le nœud pourrait ne pas fonctionner correctement ou générer une erreur. Le lien du rapport HTML généré n'est valide que lorsque le serveur local ComfyUI est en cours d'exécution.
Lien vers le Code Source du Nœud Visualize CUDAMemory History
Le nœud Visualize CUDAMemory History provient du pack de nœuds ComfyUI-KJNodes.
Commentaires
Connectez-vous avec GitHub pour rejoindre la discussion.