diff --git a/pyproject.toml b/pyproject.toml
index 620171d..a880061 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ dependencies = [
   "ruamel.yaml",
   "scmrepo>=3,<4",
   "psutil",
-  "pynvml"
+  "nvidia-ml-py"
 ]
 
 [project.optional-dependencies]