pynvml
-> nvidia-ml-py
in CI
#9111
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Attempt to resolve #9110
The warning in #9110 is about a failure to parse the warning to parse the warning from
pynvml
FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you.
This PR resolves it by following the advice of the warning and switching to the
nvidia-ml-py
package, rather than digging into the parsing error. We could look at this too, but this PR resolves the immediate issue.The
nvidia-ml-py
package also provides thepynvml
library and still usesimport pynvml
but is more officially maintained than thepynvml
package on PyPI/conda-forge. Recent releases ofpynvml
just depend onnvidia-ml-py
and raise this warning when imported.pre-commit run --all-files