Pin pynvml<11.5 for 24.12 release (#74)

Work-around for dask/distributed#8961 (which can be fixed for 25.02, but not 24.12) --------- Co-authored-by: jakirkham <[email protected]>
rapidsai · Dec 11, 2024 · a3381d0 · a3381d0
1 parent e47ba69
commit a3381d0
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" }