Skip to content

Commit

Permalink
Use PyNVML 12 (#1100)
Browse files Browse the repository at this point in the history
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: #1100
  • Loading branch information
jakirkham authored Dec 20, 2024
1 parent c7848de commit f8e34c3
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit f8e34c3

Please sign in to comment.