[Misc] Print triton info in collect_env.py (#6298)
### What this PR does / why we need it?
[Misc] Print triton info in collect_env.py, help us to collect more info
when user create an issue.
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.14.1
- vLLM main:
dc917cceb8
Signed-off-by: menogrey <1299267905@qq.com>
This commit is contained in:
@@ -67,6 +67,7 @@ DEFAULT_CONDA_PATTERNS = {
|
|||||||
"transformers",
|
"transformers",
|
||||||
"zmq",
|
"zmq",
|
||||||
"pynvml",
|
"pynvml",
|
||||||
|
"triton-ascend",
|
||||||
}
|
}
|
||||||
|
|
||||||
DEFAULT_PIP_PATTERNS = {
|
DEFAULT_PIP_PATTERNS = {
|
||||||
@@ -79,6 +80,7 @@ DEFAULT_PIP_PATTERNS = {
|
|||||||
"transformers",
|
"transformers",
|
||||||
"zmq",
|
"zmq",
|
||||||
"pynvml",
|
"pynvml",
|
||||||
|
"triton-ascend",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user