Sử dụng GPU của thẻ NVIDIA bằng công cụ Linux dstat

• 11 Tháng hai, 2017
• Vasilis Vryniotis
• . 6 Bình luận

Sản phẩm dstat là một công cụ nhỏ tuyệt vời cho phép bạn thống kê tài nguyên cho hộp Linux của mình. Nó có kiến ​​trúc mô-đun cho phép bạn phát triển các plugin bổ sung và rất dễ sử dụng. Gần đây, tôi đã lập hồ sơ về một đường dẫn Deep Learning được phát triển với Keras và Tensorflow và tôi cần thống kê chi tiết về việc sử dụng CPU, Đĩa cứng và GPU. Hai cái đầu tiên có sẵn bởi dstat, tuy nhiên theo tôi biết thì không có plugin nào để theo dõi việc sử dụng GPU cho cạc đồ họa NVIDIA.

Rất may, rất dễ dàng để viết một plugin python cho dstat. Tôi đã gửi một yêu cầu kéo trên repo chính thức nhưng vì các phiên bản mới được phát hành tương đối hiếm, đây là một số hướng dẫn về cách thiết lập plugin sử dụng GPU dstat NVIDIA trên hộp của bạn.

của DINTEK

Các lệnh sau được thử nghiệm trên Ubuntu 16.04 và chúng sẽ giúp bạn cài đặt dstat, Python NVIDIA Management Library và plugin dstat nvidia của tôi:

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

Để nhận tất cả các thống kê mặc định cùng với mức sử dụng GPU (phần trăm), hãy nhập lệnh sau:

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

Để có được tất cả thống kê sử dụng cho từng GPU, hãy sử dụng lệnh sau:

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

Cách thức thực hiện

Plugin tìm nạp số lượng GPU có sẵn trên hệ thống và lấy mẫu gấp 10 lần số liệu sử dụng cho mỗi GPU. Lấy mẫu nhiều lần hy vọng sẽ trả về các chỉ số mượt mà hơn là lấy một phép đo duy nhất. Sau đó, nó tính trung bình mức sử dụng trên tất cả các GPU và trả về kết quả cho người dùng. Mã nguồn của plugin có sẵn tại đây.

Hy vọng bạn sẽ thích nó, vui vẻ lập trình GPU! 🙂