Skip to content

Commit

Permalink
Use PyNVML 11 for now
Browse files Browse the repository at this point in the history
PyNVML 12 will depend on nvidia-ml-py and drop its equivalent
functionality. So it will largely become a metapackage with some
utilities. This will mean using a much newer version of the NVML
functionality through Python than is currently used in RAPIDS. As a
preventive step, stick to PyNVML 11, which is what RAPIDS already uses.

Then when PyNVML 12 is released, this can be relaxed on a
project-by-project basis to see if it unconvers any issues that will
need project-level fixes.
  • Loading branch information
jakirkham committed Nov 26, 2024
1 parent 46430e6 commit 305ddab
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion conda/environments/all_arch-any.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ dependencies:
- nodejs=18
- pre-commit
- psutil
- pynvml
- pynvml>=11.0.0,<12.0.0a0
- pytest
- pytest-asyncio
- pytest-jupyter[server]>=0.6.0
Expand Down
2 changes: 1 addition & 1 deletion conda/recipes/jupyterlab-nvdashboard/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ requirements:
run:
- python >=3.8
- jupyterlab >=4
- pynvml
- pynvml >=11.0.0,<12.0.0a0
- psutil


Expand Down
2 changes: 1 addition & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ dependencies:
- output_types: [conda, requirements, pyproject]
packages:
- jupyterlab>=4
- pynvml
- pynvml>=11.0.0,<12.0.0a0
- psutil
test_python:
common:
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 @@ classifiers = [
dependencies = [
"jupyterlab>=4",
"psutil",
"pynvml",
"pynvml>=11.0.0,<12.0.0a0",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
dynamic = ["version", "description", "authors", "urls", "keywords"]

Expand Down

0 comments on commit 305ddab

Please sign in to comment.