Mendapatkan penggunaan GPU kartu NVIDIA dengan alat Linux dstat

• Februari 11, 2017
• Vasilis Vriniotis
• . 6 Komentar

Grafik dstat adalah alat kecil yang luar biasa yang memungkinkan Anda untuk mendapatkan statistik sumber daya untuk kotak Linux Anda. Ini memiliki arsitektur modular yang memungkinkan Anda untuk mengembangkan plugin tambahan dan mudah digunakan. Baru-baru ini saya membuat profil saluran pipa Deep Learning yang dikembangkan dengan Keras dan Tensorflow dan saya membutuhkan statistik terperinci tentang penggunaan CPU, Hard Disk, dan GPU. Dua yang pertama tersedia di luar kotak oleh dstat, namun sejauh yang saya tahu tidak ada plugin untuk memantau penggunaan GPU untuk kartu grafis NVIDIA.

Untungnya sangat mudah untuk menulis plugin python untuk dstat. Saya sudah mengirim permintaan tarik pada repo resmi tetapi karena versi baru dirilis relatif jarang di sini adalah beberapa instruksi tentang cara mengatur plugin penggunaan dstat NVIDIA GPU di kotak Anda.

Instalasi

Perintah-perintah berikut ini diuji pada Ubuntu 16.04 dan mereka akan membantu Anda menginstal dstat, Python NVIDIA Management Library dan plugin dstat nvidia saya:

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

Untuk mendapatkan semua statistik default beserta penggunaan (persentase) GPU, ketik perintah berikut:

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

Untuk mendapatkan semua statistik penggunaan untuk setiap GPU gunakan perintah berikut:

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

Cara kerjanya

Plugin mengambil jumlah GPU yang tersedia pada sistem dan sampel 10 kali metrik penggunaan untuk setiap GPU. Pengambilan sampel beberapa kali semoga akan mengembalikan metrik yang lebih lancar daripada mendapatkan pengukuran tunggal. Setelah itu rata-rata penggunaan di semua GPU dan mengembalikan hasilnya kepada pengguna. Kode sumber plugin tersedia di sini.

Semoga Anda menikmatinya, selamat pemrograman GPU! 🙂