diff --git a/conda/recipes/rapids-dask-dependency/meta.yaml b/conda/recipes/rapids-dask-dependency/meta.yaml
index 4b8ca07..8c1c505 100644
--- a/conda/recipes/rapids-dask-dependency/meta.yaml
+++ b/conda/recipes/rapids-dask-dependency/meta.yaml
@@ -32,6 +32,7 @@ requirements:
     - dask-core ==2024.11.2
     - distributed ==2024.11.2
     - dask-expr ==1.1.19
+    - pynvml >=11.0.0,<11.5.0a0
 
 about:
   home: https://rapids.ai/
diff --git a/pyproject.toml b/pyproject.toml
index 47b8f77..b0527d3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "dask==2024.11.2",
     "distributed==2024.11.2",
     "dask-expr==1.1.19",
+    "pynvml>=11.0.0,<11.5.0a0",
 ]
 license = { text = "Apache 2.0" }
 readme = { file = "README.md", content-type = "text/markdown" }