From c34ebdd416e71a5ec5441d1356803d736777642d Mon Sep 17 00:00:00 2001 From: jakirkham Date: Wed, 18 Dec 2024 20:04:10 -0800 Subject: [PATCH] Use PyNVML 12 --- conda/environments/all_cuda-114_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +- dependencies.yaml | 2 +- pyproject.toml | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/conda/environments/all_cuda-114_arch-x86_64.yaml b/conda/environments/all_cuda-114_arch-x86_64.yaml index b1e2e491..713d251c 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,<12.0.0a0 +- pynvml>=12.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 3144dee4..518f86dd 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,<12.0.0a0 +- pynvml>=12.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 95bb356c..5cdcab0a 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,<12.0.0a0 +- pynvml>=12.0.0,<13.0.0a0 - pytest - pytest-cov - python>=3.10,<3.13 diff --git a/dependencies.yaml b/dependencies.yaml index 08f5ce2f..7b4d9434 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,<12.0.0a0 + - pynvml>=12.0.0,<13.0.0a0 - rapids-dask-dependency==25.2.*,>=0.0.0a0 - zict>=2.0.0 test_python: diff --git a/pyproject.toml b/pyproject.toml index 01c8d956..7469a345 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,<12.0.0a0", + "pynvml>=12.0.0,<13.0.0a0", "rapids-dask-dependency==25.2.*,>=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`.