20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[misc] add nvidia related library in collect env (#7674)

Browse Source
This commit is contained in:
youkaichao 2024-08-19 23:22:49 -07:00 committed by GitHub
parent 3d8a5f063d
commit c4be16e1a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
collect_env.py
View File

@ -66,6 +66,8 @@ DEFAULT_CONDA_PATTERNS = {
"nccl", "nccl",
"transformers", "transformers",
"zmq", "zmq",
"nvidia",
"pynvml",
} }
DEFAULT_PIP_PATTERNS = { DEFAULT_PIP_PATTERNS = {
@ -79,6 +81,8 @@ DEFAULT_PIP_PATTERNS = {
"nccl", "nccl",
"transformers", "transformers",
"zmq", "zmq",
"nvidia",
"pynvml",
} }