Skip to content

Commit 70c094a

Browse files
authored
[misc][cuda] improve pynvml warning (vllm-project#7852)
1 parent 2059b8d commit 70c094a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/platforms/cuda.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,9 @@
2121
if pynvml.__file__.endswith("__init__.py"):
2222
logger.warning(
2323
"You are using a deprecated `pynvml` package. Please install"
24-
" `nvidia-ml-py` instead. See https://pypi.org/project/pynvml "
24+
" `nvidia-ml-py` instead, and make sure to uninstall `pynvml`."
25+
" When both of them are installed, `pynvml` will take precedence"
26+
" and cause errors. See https://pypi.org/project/pynvml "
2527
"for more information.")
2628

2729
# NVML utils

0 commit comments

Comments
 (0)