OpenVINO add VRAM size

vladmandic · Sep 28, 2023 · 66a6d6b · 66a6d6b
1 parent 75b6c5a
commit 66a6d6b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/scripts/system-info.py b/scripts/system-info.py
@@ -191,6 +191,16 @@ def gb(val: float):
             mem.update({ 'utilization': torch.cuda.utilization() }) # do this one separately as it may fail
         except Exception:
             pass
+    else:
+        try:
+            from openvino.runtime import Core as OpenVINO_Core
+            from modules.intel.openvino import get_device as get_raw_openvino_device
+            openvino_core = OpenVINO_Core()
+            mem.update({
+                'gpu': { 'total': gb(openvino_core.get_property(get_raw_openvino_device(), 'GPU_DEVICE_TOTAL_MEM_SIZE')) },
+            })
+        except Exception:
+            pass
     return mem