不要直接用 transformers + PyTorch 裸跑,性能和显存都很差。
| 框架 | 特点 | 适用 |
|---|---|---|
| llama.cpp | CPU/GPU 混合、量化好、极快 | 本地 / 边缘 |
| vLLM | 高吞吐、批处理强 | 服务化 |
| TGI (Text Generation Inference) | 生产级 | 企业 |
| Ollama | 易用,底层 llama.cpp | 快速试用 |
✅ 性能排序(一般):
llama.cpp / vLLM > TGI > transformersLlama 3 官方是 FP16,显存压力大。
示例:
# llama.cpp
./main -m llama3-8b.Q4_K_M.gguf -ngl 35✅ 8B 模型:
nvidia-smi确保:
-ngl 35 # 层数 offload 到 GPUvllm serve meta-llama/Llama-3-8b \
--tensor-parallel-size 1 \
--gpu-memory-utilization 0.9sudo swapoff -aulimit -n 65535echo 1024 | sudo tee /proc/sys/vm/nr_hugepagessudo cpupower frequency-set -g performancemax_num_seqs--max-num-seqs 256server 模式./server -m llama3.Q4_K_M.gguf -ngl 35mmap(llama.cpp 默认)Ollama + Q4vLLM + AWQ + A100llama.cpp + Q4 + GPU offload| 现象 | 原因 |
|---|---|
| 速度慢 | 模型在 CPU |
| 显存溢出 | 未量化 |
| 吞吐低 | 未开 batch |
| 延迟高 | swap 开启 |
如果你愿意,可以告诉我:
我可以给你 具体命令级配置方案。