Use PyNVML 12 (#345)

Bump `pynvml` from `11` to `12`. This version of `pynvml` also now depends on `nvidia-ml-py` for core functionality.

Authors:
  - https://github.com/jakirkham

Approvers:
  - James Lamb (https://github.com/jameslamb)
  - Peter Andreas Entschev (https://github.com/pentschev)

URL: #345

conda/environments/all_cuda-118_arch-x86_64.yaml

@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1,<12.0.0a0
+- pynvml>=12.0.0,<13.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*

conda/environments/all_cuda-125_arch-x86_64.yaml

@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1,<12.0.0a0
+- pynvml>=12.0.0,<13.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*

conda/recipes/ucxx/meta.yaml

@@ -239,8 +239,8 @@ outputs:
         - ucx >=1.15.0,<1.18.0
         - {{ pin_subpackage('libucxx', exact=True) }}
         - {{ pin_compatible('rmm', max_pin='x.x') }}
-        - numpy>=1.23,<3.0a0
-        - pynvml >=11.4.1,<12.0.0a0
+        - numpy >=1.23,<3.0a0
+        - pynvml >=12.0.0,<13.0.0a0
       run_constrained:
         - cupy >=9.5.0
         - numba >=0.57.0

dependencies.yaml

@@ -283,7 +283,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - &numpy numpy>=1.23,<3.0a0
-          - pynvml>=11.4.1,<12.0.0a0
+          - pynvml>=12.0.0,<13.0.0a0
   run_python_distributed_ucxx:
     common:
       - output_types: [conda, requirements, pyproject]

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,<12.0.0a0",
+    "pynvml>=12.0.0,<13.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 = [