Add a 'vram usage' function that works across GPUs on Windows; It is slow, so is only useful for non critical paths

Author: cmdr2

sdkit/utils/__init__.py

@@ -30,6 +30,7 @@
 from .memory_utils import (
     gc,
     get_device_usage,
+    get_vram_usage_slow,
     get_object_id,
     get_tensors_in_memory,
     print_largest_tensors_in_memory,

sdkit/utils/memory_utils.py

@@ -57,6 +57,18 @@ def get_device_usage(device, log_info=False, process_usage_only=True, log_prefix
     return cpu_used, ram_used, ram_total, vram_used, vram_total, vram_peak
 
 
+def get_vram_usage_slow():
+    import subprocess
+    import platform
+
+    if platform.system() == "Windows":
+        ps_cmd = "(Get-Counter -Counter '\GPU Adapter Memory(*)\Dedicated Usage').CounterSamples[0].CookedValue"
+        x = subprocess.check_output(["powershell", "-Command", ps_cmd], encoding="utf-8")
+        return int(x)
+
+    return 0
+
+
 def get_object_id(o):
     """
     Returns a more-readable object id, than the long number returned by the inbuilt `id()` function.