[misc][cuda] improve pynvml warning (#7852)

70c094ad · youkaichao · GitHub · 2059b8d9 · 70c094ad
Unverified Commit 70c094ad authored Aug 25, 2024 by youkaichao Committed by GitHub Aug 25, 2024
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

vllm/platforms/cuda.py vllm/platforms/cuda.py +3 -1

No files found.
--- 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