Merge pull request #348 from KTibow/patch-5

modernize pynvml dep

Author: jsuarez5341

setup.py

@@ -258,7 +258,7 @@ def run(self):
     install_requires += [
         'torch',
         'psutil',
-        'pynvml',
+        'nvidia-ml-py',
         'rich',
         'rich_argparse',
         'imageio',