[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

View File

@ -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