update nvml

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 17:24:32 +02:00 · 2025-02-01 10:36:12 -05:00
parent 19b01e5000
commit ceaf0238fa
3 changed files with 7 additions and 1 deletions
@@ -86,6 +86,9 @@ def get_nvml():


 if __name__ == '__main__':
+    nvml_initialized = True
+    import pynvml # pylint: disable=redefined-outer-name
+    pynvml.nvmlInit()
    from rich import print as rprint
    for gpu in get_nvml():
        rprint(gpu)