Uporaba GPU kartice NVIDIA z orodjem dstat Linux

• Februar 11, 2017
• Vasilis Vryniotis
• . 6 komentarjev

O dstat je super majhno orodje, ki vam omogoča pridobivanje statističnih podatkov o virih za vaš Linux box. Ima modularno arhitekturo, ki vam omogoča razvoj dodatnih vtičnikov in je enostaven za uporabo. Pred kratkim sem profiliral cevovod za globoko učenje, razvit s Keras in Tensorflow, in potreboval sem podrobne statistične podatke o uporabi procesorja, trdega diska in grafičnega procesorja. Prva dva sta na voljo takoj po izdelavi s strani dstat, kljub temu pa, kolikor vem, ni vtičnika za spremljanje uporabe GPE za grafične kartice NVIDIA.

Na srečo je zelo enostavno napisati vtičnik python za dstat. Na uradni repo sem že poslal zahtevo za vleko, a ker so nove različice izdane razmeroma redko, je tukaj nekaj navodil za nastavitev vtičnik za uporabo dstat NVIDIA GPU na svoji škatli.

namestitev

Naslednji ukazi so preizkušeni v Ubuntu 16.04 in vam bodo pomagali namestiti dstat, knjižnico upravljanja Python NVIDIA in moj vtičnik 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

Če želite pridobiti vse privzete statistične podatke skupaj z uporabo GPE (odstotek), vnesite naslednji ukaz:

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

Če želite pridobiti vse statistične podatke o uporabi za vsak GPE, uporabite naslednji ukaz:

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

Kako deluje

Vtičnik pridobi število razpoložljivih grafičnih procesorjev v sistemu in vzorči 10-kratno meritev uporabe za vsak grafični procesor. Upamo, da bo večkratno vzorčenje vrnilo bolj gladke meritve kot ena sama meritev. Nato izračuna povprečje uporabe v vseh grafičnih procesorjih in vrne rezultate uporabniku. Izvorna koda vtičnika je na voljo tukaj.

Upam, da boste uživali, veselo programiranje GPE! 🙂