Use PyNVML 12

rapidsai · Dec 19, 2024 · 518509b · 518509b
1 parent c7848de
commit 518509b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -137,7 +137,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - numpy>=1.23,<3.0a0
-          - pynvml>=11.4.1,<12.0.0a0
+          - pynvml>=12.0.0,<13.0.0a0
   depends_on_cupy:
     common:
       - output_types: conda

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ license = { text = "BSD-3-Clause" }
 requires-python = ">=3.10"
 dependencies = [
     "numpy>=1.23,<3.0a0",
-    "pynvml>=11.4.1,<12.0.0a0",
+    "pynvml>=12.0.0,<13.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [
     "Intended Audience :: Developers",