Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
ad28a74b
Unverified
Commit
ad28a74b
authored
Aug 20, 2024
by
youkaichao
Committed by
GitHub
Aug 20, 2024
Browse files
[misc][cuda] add warning for pynvml user (#7675)
parent
e6d811dd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
vllm/platforms/cuda.py
vllm/platforms/cuda.py
+6
-0
No files found.
vllm/platforms/cuda.py
View file @
ad28a74b
...
@@ -18,6 +18,12 @@ logger = init_logger(__name__)
...
@@ -18,6 +18,12 @@ logger = init_logger(__name__)
_P
=
ParamSpec
(
"_P"
)
_P
=
ParamSpec
(
"_P"
)
_R
=
TypeVar
(
"_R"
)
_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
# NVML utils
# Note that NVML is not affected by `CUDA_VISIBLE_DEVICES`,
# Note that NVML is not affected by `CUDA_VISIBLE_DEVICES`,
# all the related functions work on real physical device ids.
# all the related functions work on real physical device ids.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment