รับการใช้งาน GPU ของการ์ด NVIDIA ด้วยเครื่องมือ Linux dstat

• กุมภาพันธ์ 11, 2017
• Vasilis Vryniotiso
• . 6 ความคิดเห็น

พื้นที่ dstat เป็นเครื่องมือเล็ก ๆ ที่ยอดเยี่ยมที่ช่วยให้คุณได้รับสถิติทรัพยากรสำหรับกล่อง Linux ของคุณ มีสถาปัตยกรรมแบบแยกส่วนซึ่งช่วยให้คุณพัฒนาปลั๊กอินเพิ่มเติมและใช้งานง่าย เมื่อเร็ว ๆ นี้ฉันกำลังสร้างโปรไฟล์ Deep Learning ไปป์ไลน์ที่พัฒนาด้วย Keras และ Tensorflow และฉันต้องการสถิติโดยละเอียดเกี่ยวกับการใช้งาน CPU, Hard Disk และ GPU สองตัวแรกมีให้ใช้งานโดย dstat แต่เท่าที่ฉันรู้ว่าไม่มีปลั๊กอินสำหรับตรวจสอบการใช้งาน GPU สำหรับการ์ดกราฟิก NVIDIA

โชคดีที่การเขียนปลั๊กอิน python สำหรับ dstat นั้นง่ายมาก ฉันได้ส่งคำขอดึงไปที่ repo อย่างเป็นทางการแล้ว แต่เนื่องจากมีการเปิดตัวเวอร์ชันใหม่ค่อนข้างน้อย นี่คือคำแนะนำบางประการเกี่ยวกับวิธีตั้งค่า ปลั๊กอินการใช้งาน dstat NVIDIA GPU บนกล่องของคุณ

การติดตั้ง

คำสั่งต่อไปนี้ได้รับการทดสอบบน Ubuntu 16.04 และจะช่วยให้คุณติดตั้ง dstat, Python NVIDIA Management Library และปลั๊กอิน 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 ที่มีความสุข! 🙂