diff --git a/pyproject.toml b/pyproject.toml
index deec3c6..3c77d24 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,6 +29,7 @@ dependencies = [
     "psutil>=6.1.0", 
     "pydantic>=2.10",
     "pyyaml>=6.0",
+    "pynvml>=11.0.0",
 ]
 
 [project.optional-dependencies]