Skip to content

Replace pynvml with nvidia-ml-py in requirements#4118

Merged
lvhan028 merged 4 commits intoInternLM:mainfrom
myhloli:remove_pynvml
Nov 11, 2025
Merged

Replace pynvml with nvidia-ml-py in requirements#4118
lvhan028 merged 4 commits intoInternLM:mainfrom
myhloli:remove_pynvml

Conversation

@myhloli
Copy link
Copy Markdown
Contributor

@myhloli myhloli commented Nov 11, 2025

/lib/python3.12/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]

@jinminxi104 jinminxi104 requested a review from grimoire November 11, 2025 08:29
Added nvidia-ml-py to the runtime dependencies.
@lvhan028
Copy link
Copy Markdown
Collaborator

Hi, @myhloli
Thank you for bringing this to our attention.
You can go ahead and remove pynvml from runtime_cuda.txt as LMDeploy no longer relies on it.
Additionally, would you be willing to remove it from the unit-test.yml file as well? We'd really appreciate your help with that.

@myhloli
Copy link
Copy Markdown
Contributor Author

myhloli commented Nov 11, 2025

Okay, after I directly uninstalled pynvml and nvidia-ml-py locally, the program runs normally without any exceptions.

@lvhan028 lvhan028 merged commit 1673861 into InternLM:main Nov 11, 2025
4 of 5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants