diff --git a/conda/environments/all_cuda-114_arch-x86_64.yaml b/conda/environments/all_cuda-114_arch-x86_64.yaml index 4a6e2cd5..a1540b5a 100644 --- a/conda/environments/all_cuda-114_arch-x86_64.yaml +++ b/conda/environments/all_cuda-114_arch-x86_64.yaml @@ -20,7 +20,7 @@ dependencies: - numpydoc>=1.1.0 - pandas>=1.3 - pre-commit -- pynvml>=11.0.0 +- pynvml>=11.0.0,<13.0.0a0 - pytest - pytest-cov - python>=3.10,<3.13 diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 6b6a637b..32858e60 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -20,7 +20,7 @@ dependencies: - numpydoc>=1.1.0 - pandas>=1.3 - pre-commit -- pynvml>=11.0.0 +- pynvml>=11.0.0,<13.0.0a0 - pytest - pytest-cov - python>=3.10,<3.13 diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index f97f3771..3b9be554 100644 --- a/conda/environments/all_cuda-125_arch-x86_64.yaml +++ b/conda/environments/all_cuda-125_arch-x86_64.yaml @@ -21,7 +21,7 @@ dependencies: - numpydoc>=1.1.0 - pandas>=1.3 - pre-commit -- pynvml>=11.0.0 +- pynvml>=11.0.0,<13.0.0a0 - pytest - pytest-cov - python>=3.10,<3.13 diff --git a/dependencies.yaml b/dependencies.yaml index 47c1a71a..e9c229c7 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -157,7 +157,7 @@ dependencies: - numba>=0.57 - numpy>=1.23,<3.0a0 - pandas>=1.3 - - pynvml>=11.0.0 + - pynvml>=11.0.0,<13.0.0a0 - rapids-dask-dependency==24.12.*,>=0.0.0a0 - zict>=2.0.0 test_python: diff --git a/pyproject.toml b/pyproject.toml index 11802a1e..8213e4c5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,7 +20,7 @@ dependencies = [ "numba>=0.57", "numpy>=1.23,<3.0a0", "pandas>=1.3", - "pynvml>=11.0.0", + "pynvml>=11.0.0,<13.0.0a0", "rapids-dask-dependency==24.12.*,>=0.0.0a0", "zict>=2.0.0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.