Skip to content

[CI] replace deprecated pynvml with nvidia-ml-py #7290

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ooooo-create wants to merge 7 commits into
base: develop
Choose a base branch
from
Open

[CI] replace deprecated pynvml with nvidia-ml-py #7290

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
ee8762a
chore: replace deprecated pynvml with nvidia-ml-py
ooooo-create Apr 10, 2026
5fd37a0
docs: remove duplicated nvidia-ml-py entries in collect-env docs
ooooo-create Apr 10, 2026
94fa594
trigger ci
ooooo-create Apr 10, 2026
d1ced44
Merge branch 'develop' into fix-pynvml-deprecated
ooooo-create Apr 15, 2026
024fd69
keep collect_env compatible with pynvml
ooooo-create Apr 21, 2026
41980ee
Merge branch 'develop' into fix-pynvml-deprecated
ooooo-create Apr 21, 2026
1055eaa
Merge branch 'develop' into fix-pynvml-deprecated
ooooo-create Apr 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions docs/cli/collect-env.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -202,7 +202,6 @@ Versions of relevant libraries
[pip3] onnx==1.18.0
[pip3] onnxoptimizer==0.3.13
[pip3] paddle2onnx==2.0.1
[pip3] pynvml==12.0.0
[pip3] pyzmq==26.4.0
Comment thread
ooooo-create marked this conversation as resolved.
[pip3] torch==2.5.1+cu118
[pip3] torchaudio==2.5.1+cu118
Expand Down Expand Up @@ -240,7 +239,6 @@ Versions of relevant libraries
[conda] nvidia-nvjitlink-cu12 12.6.85 pypi_0 pypi
[conda] nvidia-nvtx-cu11 11.8.86 pypi_0 pypi
[conda] nvidia-nvtx-cu12 12.6.77 pypi_0 pypi
[conda] pynvml 12.0.0 pypi_0 pypi
[conda] pyzmq 26.4.0 pypi_0 pypi
Comment thread
ooooo-create marked this conversation as resolved.
[conda] torch 2.5.1+cu118 pypi_0 pypi
[conda] torchaudio 2.5.1+cu118 pypi_0 pypi
Expand Down
2 changes: 0 additions & 2 deletions docs/zh/cli/collect-env.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,6 @@ Versions of relevant libraries
[pip3] onnx==1.18.0
[pip3] onnxoptimizer==0.3.13
[pip3] paddle2onnx==2.0.1
[pip3] pynvml==12.0.0
[pip3] pyzmq==26.4.0
Comment thread
ooooo-create marked this conversation as resolved.
[pip3] torch==2.5.1+cu118
[pip3] torchaudio==2.5.1+cu118
Expand Down Expand Up @@ -232,7 +231,6 @@ Versions of relevant libraries
[conda] nvidia-nvjitlink-cu12 12.6.85 pypi_0 pypi
[conda] nvidia-nvtx-cu11 11.8.86 pypi_0 pypi
[conda] nvidia-nvtx-cu12 12.6.77 pypi_0 pypi
[conda] pynvml 12.0.0 pypi_0 pypi
[conda] pyzmq 26.4.0 pypi_0 pypi
Comment thread
ooooo-create marked this conversation as resolved.
[conda] torch 2.5.1+cu118 pypi_0 pypi
[conda] torchaudio 2.5.1+cu118 pypi_0 pypi
Expand Down
4 changes: 2 additions & 2 deletions fastdeploy/collect_env.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@
"transformers",
"zmq",
"nvidia",
"pynvml",
"nvidia-ml-py",
}
Comment thread
ooooo-create marked this conversation as resolved.

DEFAULT_PIP_PATTERNS = {
Expand All @@ -107,7 +107,7 @@
"transformers",
"zmq",
"nvidia",
Comment thread
ooooo-create marked this conversation as resolved.
"pynvml",
"nvidia-ml-py",
}


Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ zmq
aiozmq
openai>=1.93.0
tqdm
pynvml
nvidia-ml-py
uvicorn>=0.38.0
fastapi
paddleformers>=1.1.1
Expand Down
2 changes: 1 addition & 1 deletion requirements_dcu.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ zmq
aiozmq
openai
tqdm
pynvml
nvidia-ml-py
uvicorn>=0.38.0
fastapi
paddleformers==1.0.0
Expand Down
2 changes: 1 addition & 1 deletion requirements_iluvatar.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ zmq
aiozmq
openai>=1.93.0
tqdm
pynvml
nvidia-ml-py
uvicorn>=0.38.0
fastapi
paddleformers==1.0.0
Expand Down
2 changes: 1 addition & 1 deletion requirements_metaxgpu.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ zmq
aiozmq
openai>=1.93.0
tqdm
pynvml
nvidia-ml-py
uvicorn>=0.38.0
fastapi
paddleformers==0.4.1
Expand Down
Loading