Przeglądaj źródła

Use official nvidia-ml-py instead of pynvml

Gaetan Lepage 7 miesięcy temu
rodzic
commit
9ffd8116f0
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      setup.py

+ 1 - 1
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",