You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The following warning info can be hidden by switching the package
`pynvml` to `nvidia-ml-py`.
```
.venv/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]
```
The Usage remains the same.
Ref: https://pypi.org/project/pynvml/ and
https://pypi.org/project/nvidia-ml-py/
Relevant reproducible detail:
```
$ uv pip list | grep -E "(torch|flashinfer)"
flashinfer-python 0.3.1
torch 2.7.1
torchaudio 2.7.1
torchvision 0.22.1
```
0 commit comments