louis_lifu
/
vllm
mirror of https://github.com/vllm-project/vllm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
youkaichao 2024-08-25 14:30:09 -07:00 committed by GitHub
parent 2059b8d9ca
commit 70c094ade6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/platforms/cuda.py
View File

@ -21,7 +21,9 @@ _R = TypeVar("_R")
if pynvml.__file__.endswith("__init__.py"): if pynvml.__file__.endswith("__init__.py"):
logger.warning( logger.warning(
"You are using a deprecated `pynvml` package. Please install" "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.") "for more information.")
# NVML utils # NVML utils