From a3381d026051853727343b1a875839a7b3784e25 Mon Sep 17 00:00:00 2001 From: "Richard (Rick) Zamora" Date: Wed, 11 Dec 2024 08:19:55 -0600 Subject: [PATCH] Pin pynvml<11.5 for 24.12 release (#74) Work-around for https://github.com/dask/distributed/issues/8961 (which can be fixed for 25.02, but not 24.12) --------- Co-authored-by: jakirkham --- conda/recipes/rapids-dask-dependency/meta.yaml | 1 + pyproject.toml | 1 + 2 files changed, 2 insertions(+) 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" }