diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 62d6d650..ce5efad9 100644 --- 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 index 4423d914..3b853e64 100644 --- 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/conda/recipes/ucxx/meta.yaml b/conda/recipes/ucxx/meta.yaml index 8d9e3f13..3b391983 100644 --- a/conda/recipes/ucxx/meta.yaml +++ b/conda/recipes/ucxx/meta.yaml @@ -240,7 +240,7 @@ outputs: - {{ pin_subpackage('libucxx', exact=True) }} - {{ pin_compatible('rmm', max_pin='x.x') }} - numpy>=1.23,<3.0a0 - - pynvml >=11.4.1 + - pynvml >=11.4.1,<12.0.0a0 run_constrained: - cupy >=9.5.0 - numba >=0.57.0 diff --git a/dependencies.yaml b/dependencies.yaml index 57261ef0..738935e4 100644 --- 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 index 2f09fa9e..55dd719e 100644 --- a/python/ucxx/pyproject.toml +++ b/python/ucxx/pyproject.toml @@ -21,7 +21,7 @@ requires-python = ">=3.10" dependencies = [ "libucxx==0.42.*,>=0.0.0a0", "numpy>=1.23,<3.0a0", - "pynvml>=11.4.1", + "pynvml>=11.4.1,<12.0.0a0", "rmm==25.2.*,>=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 = [