Obtendo o uso da GPU das placas NVIDIA com a ferramenta dstat do Linux

• 11 de fevereiro de 2017
• Vasilis Vryniotis
• . 6 Comentários

A dstat é uma pequena ferramenta incrível que permite que você obtenha estatísticas de recursos para sua caixa Linux. Possui uma arquitetura modular que permite desenvolver plugins adicionais e é fácil de usar. Recentemente, eu estava criando o perfil de um pipeline de Deep Learning desenvolvido com Keras e Tensorflow e precisava de estatísticas detalhadas sobre o uso de CPU, disco rígido e GPU. Os dois primeiros estão disponíveis prontos para uso pelo dstat, no entanto, até onde eu sei, não há plug-in para monitorar o uso da GPU para placas gráficas NVIDIA.

Felizmente, é super fácil escrever um plugin python para o dstat. Já enviei uma solicitação de pull no repositório oficial, mas como novas versões são lançadas relativamente raramente, aqui estão algumas instruções sobre como configurar o dstat plug-in de uso da GPU NVIDIA em sua caixa.

Instalação

Os comandos a seguir são testados no Ubuntu 16.04 e ajudarão você a instalar o dstat, a Python NVIDIA Management Library e meu plugin dstat nvidia:

sudo apt-get install dstat #install dstat
sudo pip install nvidia-ml-py #install Python NVIDIA Management Library
wget https://raw.githubusercontent.com/datumbox/dstat/master/plugins/dstat_nvidia_gpu.py
sudo mv dstat_nvidia_gpu.py /usr/share/dstat/ #move file to the plugins directory of dstat

Para obter todas as estatísticas padrão junto com o uso da GPU (porcentagem), digite o seguinte comando:

dstat -a --nvidia-gpu

----total-cpu-usage---- -dsk/total- -net/total- ---paging-- ---system-- gpu-u
usr sys idl wai hiq siq| read  writ| recv  send|  in   out | int   csw |total
  2   1  96   0   0   0|5816k   15M|   0     0 |   0     0 |  45k   98k|   68
  0   1  98   0   0   0|  57M  128k| 104B  902B|   0     0 |  42k   85k|   50
  8   7  84   1   0   0| 152M    0 | 292B  448B|   0     0 |  52k   93k|   39
  1   1  97   1   0   0| 111M    0 |  52B  374B|   0     0 |  51k  116k|   62
  0   1  98   1   0   0| 129M    0 |  80B  416B|   0     0 |  43k   85k|   92
  0   2  98   0   0   0|   0     0 |  52B  374B|   0     0 |  41k   83k|   81

Para obter todas as estatísticas de uso de cada GPU, use o seguinte comando:

dstat --nvidia-gpu -f
-------------------------------------------gpu-usage-nvidia------------------------------------------
total  gpu0  gpu1  gpu2  gpu3  gpu4  gpu5  gpu6  gpu7  gpu8  gpu9 gpu10 gpu11 gpu12 gpu13 gpu14 gpu15
   19    23    22    21    21    20    22    23    25    15    18    16    16    16    18    16    14
   18    21    20    18    22    21    21    22    21    15    15    14    14    14    15    16    13
   10    14     9    13     8     9    11     9    12     9     9    10    10     8     7     9     9
   18    20    22    19    21    20    21    21    22    14    15    14    15    14    15    15    15
   20    24    22    23    24    25    22    22    22    16    16    16    16    16    16    18    16
   15    21    18    19    18    17    17    16    18    14    13    13    14    13    12    11    11
   20    24    22    22    24    25    23    24    22    16    18    16    14    17    17    17    15
   19    29    18    23    21    22    21    20    21    18    16    16    18    14    14    17    17

Como Funciona

O plug-in busca o número de GPUs disponíveis no sistema e amostra 10 vezes a métrica de uso para cada GPU. A amostragem várias vezes retornará métricas mais suaves do que obter uma única medição. Depois disso, ele calcula a média do uso em todas as GPUs e retorna os resultados ao usuário. O código fonte do plugin está disponível SUA PARTICIPAÇÃO FAZ A DIFERENÇA.

Espero que gostem, boa programação de GPU! 🙂