Merge pull request #289 from GaetanLepage/pynvml

Use official nvidia-ml-py instead of pynvml

setup.py

@@ -12,11 +12,11 @@ install_requires = [
   "Jinja2==3.1.4",
   "netifaces==0.11.0",
   "numpy==2.0.0",
+  "nvidia-ml-py==12.560.30",
   "pillow==10.4.0",
   "prometheus-client==0.20.0",
   "protobuf==5.27.1",
   "psutil==6.0.0",
-  "pynvml==11.5.3",
   "requests==2.32.3",
   "rich==13.7.1",
   "safetensors==0.4.3",