Nvidia GPU info #18

lnxlink/consts.py

@@ -67,4 +67,5 @@
 - bash
 - required_restart
 - disk_usage
+- nvidia_gpu
 """

lnxlink/modules/nvidia_gpu.py

@@ -0,0 +1,24 @@
+import subprocess
+
+
+class Addon():
+    name = 'Nvidia GPU'
+    icon = 'mdi:expansion-card-variant'
+    unit = 'json'
+
+    def getInfo(self) -> dict:
+        stdout = subprocess.run(
+            ['nvidia-smi',
+             '--query-gpu=driver_version,name,fan.speed,memory.total,memory.used,memory.free,utilization.gpu,utilization.memory,temperature.gpu,power.draw,clocks.current.graphics',
+             '--format=csv,nounits'],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE).stdout.decode("UTF-8").strip().split('\n')
+        # stdout = map(int, stdout.split(','))
+        headers = stdout[0].split(', ')
+        values = stdout[1].split(', ')
+
+        gpu_status = {}
+        for num, header in enumerate(headers):
+            gpu_status[header] = values[num]
+        gpu_status['status'] = gpu_status['name']
+        return gpu_status