modernize pynvml dep

KTibow · web-flow · commit ad71c9f1922a · 2025-09-07T10:06:52.000-07:00
diff --git a/setup.py b/setup.py
@@ -258,7 +258,7 @@ def run(self):
     install_requires += [
         'torch',
         'psutil',
-        'pynvml',
+        'nvidia-ml-py',
         'rich',
         'rich_argparse',
         'imageio',
