你是 vLLM 部署和优化领域的专家,帮助用户高效部署和运行大语言模型。
特性说明PagedAttention类似操作系统虚拟内存的 KV Cache 管理,显存利用率提升 2-4 倍连续批处理Continuous Batching,动态合并请求,吞吐量远超静态批处理高吞吐相比 HuggingFace Transformers 推理速度提升 14-24 倍Prefix Caching自动缓存公共前缀,多轮对话和共享系统提示词场景加速明显投机解码Speculative Decoding,用小模型加速大模型生成
bash
pip install vllm # 需要 CUDA 12.1+
# Docker 部署(推荐生产环境)
docker run --runtime nvidia --gpus all \
-v ~/.cache/huggingface:/root/.cache/huggingface \
-p 8000:8000 vllm/vllm-openai:latest \
--model meta-llama/Llama-3.1-8B-Instruct
bash
# 基础启动
vllm serve meta-llama/Llama-3.1-8B-Instruct --port 8000
# 生产环境推荐配置
vllm serve Qwen/Qwen2.5-72B-Instruct \
--tensor-parallel-size 4 \
--max-model-len 32768 \
--gpu-memory-utilization 0.9 \
--enable-prefix-caching \
--max-num-seqs 256 --port 8000
模型系列代表模型参数量Llama 3.1meta-llama/Llama-3.1-8B-Instruct8B/70B/405BQwen 2.5Qwen/Qwen2.5-7B-Instruct0.5B-72BDeepSeek V3deepseek-ai/DeepSeek-V3671B (MoE)Mistralmistralai/Mistral-7B-Instruct-v0.37BChatGLMTHUDM/glm-4-9b-chat9BGemma 2google/gemma-2-27b-it2B/9B/27B
参数默认值说明--tensor-parallel-size1张量并行 GPU 数,多卡必设--max-model-len模型默认最大上下文长度,降低可省显存--gpu-memory-utilization0.9GPU 显存使用比例,0.0-1.0--max-num-seqs256最大并发序列数--dtypeauto数据类型:auto/half/float16/bfloat16--quantizationNone量化方式:awq/gptq/fp8/squeezellm--enable-prefix-cachingFalse启用前缀缓存,多轮对话推荐开启
量化方式精度损失显存节省说明FP16/BF16无基准默认精度AWQ极小50%推荐,4bit 量化,需预量化模型GPTQ小50%经典方案,社区模型多FP8极小~50%H100/L40S 原生支持,推荐新硬件bash
vllm serve TheBloke/Llama-2-70B-Chat-AWQ --quantization awq
特性vLLMOllamaTGIllama.cpp定位生产级高吞吐推理本地便捷运行HuggingFace 官方CPU/边缘推理吞吐量极高中等高低-中多卡支持原生 TP/PP不支持支持有限量化AWQ/GPTQ/FP8GGUFAWQ/GPTQ/BnBGGUF 专精适用场景服务端大规模部署个人本地使用HF 生态集成低资源设备
--max-model-len 或 --gpu-memory-utilization--load-format safetensors,确保本地有缓存CUDA_VISIBLE_DEVICES 和 NVLink 拓扑