`gpustat`

Un peu moins que nvidia-smi ?

Capture d'écran : gpustat -cp

REMARQUE : cela fonctionne uniquement avec les périphériques graphiques NVIDIA, aucune prise en charge AMD pour le moment. Les contributions sont les bienvenues !

Auto-Promotion : Une interface web de gpustat est disponible (en alpha) ! Consultez gpustat-web.

Installation rapide

Installer depuis PyPI :

 pip install gpustat

Si vous n'avez pas le privilège root (sudo), essayez d'installer gpustat sur l'espace de noms utilisateur : pip install --user gpustat .

Pour installer la dernière version (branche principale) via pip :

 pip install git+https://github.com/wookayin/gpustat.git@master

Configuration requise pour le pilote NVIDIA et `pynvml`

Important

NE PAS : pip install pynvml , ni inclure pynvml comme dépendance dans votre projet python. Cela ne fonctionnera pas.

Au lieu de cela : pip install nvidia-ml-py . nvidia-ml-py est la liaison Python officielle de NVIDIA pour NVML.

gpustat 1.2+ : nécessite nvidia-ml-py >= 12.535.108 (#161)
gpustat 1.0+ : nécessite le pilote NVIDIA 450.00 ou supérieur et nvidia-ml-py >= 11.450.129 .
Si votre pilote NVIDIA est trop ancien, vous pouvez utiliser des versions plus anciennes gpustat ( pip install gpustat<1.0 ). Voir #107 pour plus de détails.

Exigences Python

gpustat<1.0 : Compatible avec python 2.7 et >=3.4
gpustat 1.0 : Python >= 3.4
gpustat 1.1 : Python >= 3.6

Usage

$ gpustat

Options (Veuillez consulter gpustat --help pour plus de détails) :

--color : Force la sortie colorée (même lorsque la sortie standard n'est pas un terminal)
--no-color : Supprime la sortie colorée
-u , --show-user : Afficher le nom d'utilisateur du propriétaire du processus
-c , --show-cmd : Afficher le nom du processus
-f , --show-full-cmd : Afficher les commandes complètes et les statistiques du processeur du processus en cours d'exécution
-p , --show-pid : Afficher le PID du processus
-F , --show-fan : Afficher la vitesse du ventilateur du GPU
-e , --show-codec : Afficher l'utilisation de l'encodeur et/ou du décodeur
-P , --show-power : Afficher la consommation d'énergie et/ou la limite du GPU ( draw ou draw,limit )
-a , --show-all : Afficher toutes les propriétés du GPU ci-dessus
--id : ciblez et interrogez des GPU spécifiques uniquement avec les indices spécifiés (par exemple --id 0,1,2 )
--no-processes : Ne pas afficher les informations sur le processus (utilisateur, mémoire) (#133)
--watch , -i , --interval : Exécuté en mode montre (équivalent à watch gpustat ) si fourni. Indique l'intervalle entre les mises à jour.
--json : sortie JSON (#10)
--print-completion (bash|zsh|tcsh) : Imprime un script de complétion du shell. Voir #131 pour l'utilisation.

Conseils

Essayez gpustat --debug si quelque chose ne va pas.
Pour regarder périodiquement, essayez gpustat --watch ou gpustat -i (#41).
- Pour les anciennes versions, on peut utiliser watch --color -n1.0 gpustat --color .
L'exécution nvidia-smi daemon (privilège root requis) rendra l'interrogation des GPU beaucoup plus rapide et utilisera moins de CPU (#54).
L'ID GPU (index) affiché par gpustat (et nvidia-smi ) est l'ID de BUS PCI, tandis que CUDA utilise un ordre différent (attribue le GPU le plus rapide avec l'ID le plus bas) par défaut. Par conséquent, afin de garantir que CUDA et gpustat utilisent le même index GPU , configurez la variable d'environnement CUDA_DEVICE_ORDER sur PCI_BUS_ID (avant de définir CUDA_VISIBLE_DEVICES pour votre programme CUDA) : export CUDA_DEVICE_ORDER=PCI_BUS_ID .

Affichage par défaut

 [0] GeForce GTX Titan X | 77°C,  96 % | 11848 / 12287 MB | python/52046(11821M)

[0] : index GPU (commence à partir de 0) comme PCI_BUS_ID
GeForce GTX Titan X : Nom du GPU
77°C : Température du GPU (en Celsius)
96 % : Utilisation du GPU
11848 / 12287 MB : utilisation de la mémoire GPU (utilisée / totale)
python/... : Exécution de processus sur GPU, propriétaire/cmdline/PID (et leur utilisation de la mémoire GPU)