pin pynvml for 24.12 release

rapidsai · Dec 11, 2024 · 14fbec5 · 14fbec5
1 parent e47ba69
commit 14fbec5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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" }