Skip to content

pynvml package is deprecated #894

@dbrenes-rr

Description

@dbrenes-rr

Hey DVC Team, thanks for this awesome tool!

I got this FutureWarning while training a model with DVC Live. I believe pynvml is a DVC Live dependency. Just wanted to give a heads up in case you're not aware it's deprecated.

.venv/lib/python3.10/site-packages/torch/cuda/__init__.py:61: 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.
  import pynvml  # type: ignore[import]

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions