From 70c094ade6eb77396a309512f24ddbfafaf15b38 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Sun, 25 Aug 2024 14:30:09 -0700 Subject: [PATCH] [misc][cuda] improve pynvml warning (#7852) --- vllm/platforms/cuda.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vllm/platforms/cuda.py b/vllm/platforms/cuda.py index 84301afa..bda82d37 100644 --- a/vllm/platforms/cuda.py +++ b/vllm/platforms/cuda.py @@ -21,7 +21,9 @@ _R = TypeVar("_R") if pynvml.__file__.endswith("__init__.py"): logger.warning( "You are using a deprecated `pynvml` package. Please install" - " `nvidia-ml-py` instead. See https://pypi.org/project/pynvml " + " `nvidia-ml-py` instead, and make sure to uninstall `pynvml`." + " When both of them are installed, `pynvml` will take precedence" + " and cause errors. See https://pypi.org/project/pynvml " "for more information.") # NVML utils