xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-31 06:47:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] reformat the collect-env output (#18285)

Browse Source 
Signed-off-by: reidliu41 <reid201711@gmail.com>
Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid 2025-05-17 10:46:18 +08:00 committed by GitHub
parent 55f1a468d9
commit 60017dc841
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 51 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
vllm/collect_env.py
View File

@ -637,33 +637,50 @@ def get_env_info():
env_info_fmt = """ env_info_fmt = """
PyTorch version: {torch_version} ==============================
Is debug build: {is_debug_build} System Info
CUDA used to build PyTorch: {cuda_compiled_version} ==============================
ROCM used to build PyTorch: {hip_compiled_version} OS : {os}
GCC version : {gcc_version}
Clang version : {clang_version}
CMake version : {cmake_version}
Libc version : {libc_version}
OS: {os} ==============================
GCC version: {gcc_version} PyTorch Info
Clang version: {clang_version} ==============================
CMake version: {cmake_version} PyTorch version : {torch_version}
Libc version: {libc_version} Is debug build : {is_debug_build}
CUDA used to build PyTorch : {cuda_compiled_version}
ROCM used to build PyTorch : {hip_compiled_version}
Python version: {python_version} ==============================
Python platform: {python_platform} Python Environment
Is CUDA available: {is_cuda_available} ==============================
CUDA runtime version: {cuda_runtime_version} Python version : {python_version}
CUDA_MODULE_LOADING set to: {cuda_module_loading} Python platform : {python_platform}
GPU models and configuration: {nvidia_gpu_models}
Nvidia driver version: {nvidia_driver_version}
cuDNN version: {cudnn_version}
HIP runtime version: {hip_runtime_version}
MIOpen runtime version: {miopen_runtime_version}
Is XNNPACK available: {is_xnnpack_available}
CPU: ==============================
CUDA / GPU Info
==============================
Is CUDA available : {is_cuda_available}
CUDA runtime version : {cuda_runtime_version}
CUDA_MODULE_LOADING set to : {cuda_module_loading}
GPU models and configuration : {nvidia_gpu_models}
Nvidia driver version : {nvidia_driver_version}
cuDNN version : {cudnn_version}
HIP runtime version : {hip_runtime_version}
MIOpen runtime version : {miopen_runtime_version}
Is XNNPACK available : {is_xnnpack_available}
==============================
CPU Info
==============================
{cpu_info} {cpu_info}
Versions of relevant libraries: ==============================
Versions of relevant libraries
==============================
{pip_packages} {pip_packages}
{conda_packages} {conda_packages}
""".strip() """.strip()
@ -671,17 +688,23 @@ Versions of relevant libraries:
# both the above code and the following code use `strip()` to # both the above code and the following code use `strip()` to
# remove leading/trailing whitespaces, so we need to add a newline # remove leading/trailing whitespaces, so we need to add a newline
# in between to separate the two sections # in between to separate the two sections
env_info_fmt += "\n" env_info_fmt += "\n\n"
env_info_fmt += """ env_info_fmt += """
ROCM Version: {rocm_version} ==============================
Neuron SDK Version: {neuron_sdk_version} vLLM Info
vLLM Version: {vllm_version} ==============================
ROCM Version : {rocm_version}
Neuron SDK Version : {neuron_sdk_version}
vLLM Version : {vllm_version}
vLLM Build Flags: vLLM Build Flags:
{vllm_build_flags} {vllm_build_flags}
GPU Topology: GPU Topology:
{gpu_topo} {gpu_topo}
==============================
Environment Variables
==============================
{env_vars} {env_vars}
""".strip() """.strip()