Merge pull request #289 from GaetanLepage/pynvml

Use official nvidia-ml-py instead of pynvml
This commit is contained in:
Alex Cheema
2024-10-06 03:06:23 +04:00
committed by GitHub

View File

@@ -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",