Comment obtenir les informations sur le GPU ?

Si vous cherchez uniquement les noms des cartes vidéo de la machine, utilisez simplement :

$ nvidia-smi --list-gpus

Pour certains GPU plus récents, cette liste indique également la mémoire de chaque dispositif.

Pour trouver les bases, selon https://wiki.debian.org/NvidiaGraphicsDrivers#NVIDIA_Proprietary_Driver ,

lspci | grep VGA

Si vous avez besoin de plus de détails, consultez l'article de @knb. réponse à cette même question.

Si vous souhaitez avoir des informations simples, vous pouvez essayer de gpustat . Il est très bon et simple.

L'auteur donne les instructions d'installation suivantes :

Installer à partir de PyPI :

pip install gpustat 

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

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

Si vous n'avez pas le privilège de l'utilisateur root, essayez d'installer sur l'espace de noms de l'utilisateur : pip install --user . Notez que depuis la version 0.4, gpustat.py n'est plus un exécutable à dépendance zéro. Cependant, dans de rares cas, vous aurez besoin d'un seul exécutable script (héritage), vous pouvez aussi essayer :

wget https://raw.githubusercontent.com/wookayin/gpustat/v0.3.2/gpustat.py -O ~/.local/bin/gpustat chmod +x ~/.local/bin/gpustat    # Assuming ~/.local/bin is in your $PATH

Pour les GPU nvidia, nvidia-smi La commande est votre amie. Voir man nvidia-smi si vous aimez.

Pour lister les GPU, utilisez nvidia-smi -L ( nvidia-smi --list-gpus ), nvidia-smi -q donne des informations sur le gpu et les processus en cours.

Cette réponse suppose que vous disposez d'un serveur équipé de GPU NVIDIA. Vous avez trois possibilités :

1. Pour en avoir un bref aperçu : nvidia-smi

2. Pour obtenir un rapport détaillé : nvidia-smi -q . Vous obtiendrez plusieurs écrans d'informations détaillées si vous avez plus d'un gpu.

3. Faites un ls /proc/driver/nvidia/gpus/ . Il affichera l'emplacement du bus GPU sous forme de dossiers. Maintenant, exécutez la commande suivante pour chacun des emplacements du bus GPU. Remplir <gpu-id> avec localisation du bus : cat /proc/driver/nvidia/gpus/<gpu_id>/information