Set PyNVML upper pin

rapidsai · Nov 15, 2024 · 4d3c385 · 4d3c385
1 parent f1d98f2
commit 4d3c385
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1
+- pynvml>=11.4.1,<12.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1
+- pynvml>=11.4.1,<12.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -283,7 +283,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - &numpy numpy>=1.23,<3.0a0
-          - pynvml>=11.4.1
+          - pynvml>=11.4.1,<12.0.0a0
   run_python_distributed_ucxx:
     common:
       - output_types: [conda, requirements, pyproject]

diff --git a/python/ucxx/pyproject.toml b/python/ucxx/pyproject.toml
@@ -21,7 +21,7 @@ requires-python = ">=3.10"
 dependencies = [
     "libucxx==0.41.*,>=0.0.0a0",
     "numpy>=1.23,<3.0a0",
-    "pynvml>=11.4.1",
+    "pynvml>=11.4.1,<12.0.0a0",
     "rmm==24.12.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [