Skip to content

Commit ad28a74

Browse files
authored
[misc][cuda] add warning for pynvml user (vllm-project#7675)
1 parent e6d811d commit ad28a74

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

vllm/platforms/cuda.py

+6
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,12 @@
1818
_P = ParamSpec("_P")
1919
_R = TypeVar("_R")
2020

21+
if pynvml.__file__.endswith("__init__.py"):
22+
logger.warning(
23+
"You are using a deprecated `pynvml` package. Please install"
24+
" `nvidia-ml-py` instead. See https://pypi.org/project/pynvml "
25+
"for more information.")
26+
2127
# NVML utils
2228
# Note that NVML is not affected by `CUDA_VISIBLE_DEVICES`,
2329
# all the related functions work on real physical device ids.

0 commit comments

Comments
 (0)