Switch pynvml to nvidia-ml-py (#5529)

Fixes: #5517 Link to PyPI for nvidia-ml-py [here](https://pypi.org/project/nvidia-ml-py/) showing usage remaining the same as previous pynvml package.
microsoft · May 13, 2024 · 82ce4ae · 82ce4ae
1 parent 3a7f3aa
commit 82ce4ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,10 +1,10 @@
 hjson
 ninja
 numpy
+nvidia-ml-py
 packaging>=20.0
 psutil
 py-cpuinfo
 pydantic
-pynvml
 torch
 tqdm