Qu'est ce que tu fais d'habitude?
Il est supposé que vous utilisez un pilote GPU créé par NVIDIA.
Il semble qu'il soit implémenté dans le cadre d'un cadre, mais je ne le trouve pas, alors je vais écrire ce que je fais dans mon cas. (Si vous en avez un, dites-le moi en secret ...)
Tout d'abord, obtenez l'état de la mémoire utilisée pour chaque id en entrant ce qui suit en tant que commande shell.
nvidia-smi --query-gpu=index,memory.used --format=csv,noheader
Extrayez le résultat que vous souhaitez utiliser l'expression régulière suivante dans le résultat renvoyé.
^([0-9]+),(\s+)?([0-9]+)(\s+)?MiB$
Le processus en deux étapes ci-dessus est décrit comme du code python.
Recommended Posts