We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e6d811d commit ad28a74Copy full SHA for ad28a74
vllm/platforms/cuda.py
@@ -18,6 +18,12 @@
18
_P = ParamSpec("_P")
19
_R = TypeVar("_R")
20
21
+if pynvml.__file__.endswith("__init__.py"):
22
+ logger.warning(
23
+ "You are using a deprecated `pynvml` package. Please install"
24
+ " `nvidia-ml-py` instead. See https://pypi.org/project/pynvml "
25
+ "for more information.")
26
+
27
# NVML utils
28
# Note that NVML is not affected by `CUDA_VISIBLE_DEVICES`,
29
# all the related functions work on real physical device ids.
0 commit comments