wassname/vllm
1
0
Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-09-09 11:40:01 +08:00
Code Issues 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
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
collect_env.py
+4
View File
@@ -66,6 +66,8 @@ DEFAULT_CONDA_PATTERNS = {
"nccl",
"transformers",
"zmq",
"nvidia",
"pynvml",
}
DEFAULT_PIP_PATTERNS = {
@@ -79,6 +81,8 @@ DEFAULT_PIP_PATTERNS = {
"nccl",
"transformers",
"zmq",
"nvidia",
"pynvml",
}