Λήψη της χρήσης GPU των καρτών NVIDIA με το εργαλείο Linux dstat

• Φεβρουάριος 11, 2017
• Βασίλης Βρυνιώτης
• . 6 Σχόλια

Η dstat είναι ένα φοβερό μικρό εργαλείο που σας επιτρέπει να λαμβάνετε στατιστικά στοιχεία πόρων για το Linux box σας. Διαθέτει αρθρωτή αρχιτεκτονική που σας επιτρέπει να αναπτύξετε πρόσθετα πρόσθετα και είναι εύκολο στη χρήση. Πρόσφατα έφτιαχνα ένα προφίλ αγωγού Deep Learning που αναπτύχθηκε με Keras και Tensorflow και χρειάζομαι λεπτομερή στατιστικά στοιχεία σχετικά με τη χρήση της CPU, του σκληρού δίσκου και της GPU. Τα δύο πρώτα είναι διαθέσιμα εκτός συσκευασίας από το dstat, ωστόσο, όσο γνωρίζω, δεν υπάρχει πρόσθετο για την παρακολούθηση της χρήσης GPU για κάρτες γραφικών NVIDIA.

Ευτυχώς είναι εξαιρετικά εύκολο να γράψετε ένα πρόσθετο python για το dstat. Έχω ήδη στείλει μια αίτηση έλξης στο επίσημο repo, αλλά επειδή οι νέες εκδόσεις κυκλοφορούν σχετικά σπάνια εδώ είναι μερικές οδηγίες σχετικά με τον τρόπο ρύθμισης του dstat NVIDIA GPU plugin χρήση στο κουτί σας.

Εγκατάσταση

Οι ακόλουθες εντολές δοκιμάζονται στο Ubuntu 16.04 και θα σας βοηθήσουν να εγκαταστήσετε το dstat, τη βιβλιοθήκη διαχείρισης Python NVIDIA και την προσθήκη μου 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

Για να λάβετε όλα τα προεπιλεγμένα στατιστικά στοιχεία μαζί με τη χρήση GPU (ποσοστό) πληκτρολογήστε την ακόλουθη εντολή:

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

Για να λάβετε όλα τα στατιστικά χρήσης για κάθε GPU χρησιμοποιήστε την ακόλουθη εντολή:

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

Πώς λειτουργεί

Η προσθήκη λαμβάνει τον αριθμό των διαθέσιμων GPU στο σύστημα και τα δείγματα 10 φορές τη μέτρηση χρήσης για κάθε GPU. Η δειγματοληψία πολλές φορές θα επιστρέψει ελπίζω ομαλότερες μετρήσεις από τη λήψη μιας μόνο μέτρησης. Μετά από αυτό, ο μέσος όρος της χρήσης σε όλες τις GPU και επιστρέφει τα αποτελέσματα στον χρήστη. Ο πηγαίος κώδικας της προσθήκης είναι διαθέσιμος εδώ.

Ελπίζω να το απολαύσετε, ευτυχισμένος προγραμματισμός GPU! 🙂