Az NVIDIA kártyák GPU-használatának elérése a Linux dstat eszközzel

• Február 11, 2017
• Vasilis Vryniotis
• . 6 megjegyzés

A dstat egy nagyszerű kis eszköz, amely lehetővé teszi, hogy erőforrás-statisztikát kapjon a Linux-boxhoz. Moduláris felépítésű, amely lehetővé teszi további bővítmények fejlesztését, és könnyen használható. Nemrég egy Keras- és Tensorflow-val fejlesztett Deep Learning folyamatot készítettem, és részletes statisztikákra volt szükségem a CPU, a merevlemez és a GPU használatáról. Az első kettő már a dstat által elérhető, ennek ellenére tudomásom szerint nincs plugin a GPU használatának figyelésére az NVIDIA grafikus kártyákhoz.

Szerencsére rendkívül egyszerű python beépülő modult írni a dstathoz. Már küldtem egy lehívási kérelmet a hivatalos repóról, de mivel viszonylag ritkán adnak ki új verziót, itt van néhány utasítás a dstat NVIDIA GPU-használati bővítmény a dobozodon.

Telepítés

A következő parancsokat tesztelték az Ubuntu 16.04-en, és ezek segítenek a dstat, a Python NVIDIA Management Library és a dstat nvidia beépülő modul telepítésében:

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

Az összes alapértelmezett statisztika és a GPU-használat (százalék) megtekintéséhez írja be a következő parancsot:

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

Az egyes GPU-k összes használati statisztikájának megtekintéséhez használja a következő parancsot:

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

Hogyan működik?

A beépülő modul lekéri a rendszeren elérhető GPU-k számát, és minden GPU-hoz a használati mutató tízszeresét veszi minta. A többszöri mintavételezés remélhetőleg gördülékenyebb mutatókat eredményez, mint egyetlen mérés. Ezt követően átlagolja az összes GPU-t, és visszaküldi az eredményeket a felhasználónak. A bővítmény forráskódja elérhető itt.

Remélem tetszeni fog, boldog GPU programozást! 🙂