edited get hardware/sys information tutorial

x4nth055 · x4nth055 · commit db0b45ce8334 · 2020-06-03T12:19:41.000+02:00
diff --git a/general/sys-info/requirements.txt b/general/sys-info/requirements.txt
@@ -1 +1,3 @@
-psutil
+psutil
+tabulate
+gputil
diff --git a/general/sys-info/sys_info.py b/general/sys-info/sys_info.py
@@ -108,3 +108,32 @@ def get_size(bytes, suffix="B"):
 print(f"Total Bytes Received: {get_size(net_io.bytes_recv)}")
 
 
+# GPU information
+import GPUtil
+from tabulate import tabulate
+print("="*40, "GPU Details", "="*40)
+gpus = GPUtil.getGPUs()
+list_gpus = []
+for gpu in gpus:
+    # get the GPU id
+    gpu_id = gpu.id
+    # name of GPU
+    gpu_name = gpu.name
+    # get % percentage of GPU usage of that GPU
+    gpu_load = f"{gpu.load*100}%"
+    # get free memory in MB format
+    gpu_free_memory = f"{gpu.memoryFree}MB"
+    # get used memory
+    gpu_used_memory = f"{gpu.memoryUsed}MB"
+    # get total memory
+    gpu_total_memory = f"{gpu.memoryTotal}MB"
+    # get GPU temperature in Celsius
+    gpu_temperature = f"{gpu.temperature} °C"
+    gpu_uuid = gpu.uuid
+    list_gpus.append((
+        gpu_id, gpu_name, gpu_load, gpu_free_memory, gpu_used_memory,
+        gpu_total_memory, gpu_temperature, gpu_uuid
+    ))
+
+print(tabulate(list_gpus, headers=("id", "name", "load", "free memory", "used memory", "total memory",
+                                   "temperature", "uuid")))

-Original file line number
+Diff line change
@@ @@ -1 +1,3 @@ @@
 -psutil
 +psutil
 +tabulate
 +gputil