"tests/vscode:/vscode.git/clone" did not exist on "eaa2e51088d4daf36d47e566ad90e812f80e91b8"
Unverified Commit ad28a74b authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[misc][cuda] add warning for pynvml user (#7675)

parent e6d811dd
......@@ -18,6 +18,12 @@ logger = init_logger(__name__)
_P = ParamSpec("_P")
_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 "
"for more information.")
# NVML utils
# Note that NVML is not affected by `CUDA_VISIBLE_DEVICES`,
# all the related functions work on real physical device ids.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment