返回顶部
热门问答 更多热门问答
技术文章 更多技术文章

大语言模型引擎全解析:Transformers、vLLM、Llama.cpp、SGLang、MLX 和 Ollama

[复制链接]
链载Ai 显示全部楼层 发表于 1 小时前 |阅读模式 打印 上一主题 下一主题


本文将带你深入了解 Transformers、vLLM、Llama.cpp、SGLang、MLX 和 Ollama 这些引擎,帮助你找到最适合的工具,释放大语言模型的全部潜力!


一、Transformers 引擎:NLP领域的全能王者

开发者:Hugging Face

核心特色:作为当下最炙手可热的开源NLP库,Transformers堪称NLP领域的“瑞士军刀”,它支持数百种预训练模型,涵盖GPT、BERT、T5等知名模型,从模型加载、微调,到推理,提供了一站式解决方案。

显著优势

  • 兼容性强:完美适配PyTorch和TensorFlow,为开发者提供更多选择。

  • 生态繁荣:拥有活跃的社区,丰富的模型库与完善的文档,无论是初学者入门,还是专家深入研究,都能从中获益。

  • 应用广泛:适用于从学术研究到工业生产的各类NLP任务。

  • 适用场景:当你需要快速实现文本分类、生成、翻译等任务时,Transformers是不二之选,能助力你轻松实现NLP应用落地 。


二、vLLM 引擎:GPU推理的性能巅峰

开发者:UC Berkeley研究团队

核心特色:vLLM专注于大语言模型推理,凭借创新的内存管理技术(如PagedAttention),大幅提升GPU利用率与推理速度,堪称GPU推理的“性能怪兽”。

显著优势

  • 性能卓越:极致的推理速度,能够满足大规模部署需求。

  • 内存高效:高效的内存管理,支持更大的模型批次处理。

  • 场景适配:专为GPU优化,在高并发场景中表现出色。

    适用场景:若你需在生产环境部署大语言模型,并追求极致性能,vLLM无疑是最佳选择,它能提升模型推理速度,降低硬件成本 。


三、Llama.cpp 引擎:CPU上的轻量化先锋

开发者:社区项目

核心特色:Llama.cpp基于C++实现,专为运行Meta的LLaMA模型而生,通过优化计算与内存管理,让大模型在CPU上运行成为现实,是CPU设备上的“轻量级王者”。

显著优势

  • 轻量运行:无需GPU,在普通CPU设备上即可运行。

  • 灵活部署:适合资源受限环境,如嵌入式设备、低配服务器。

  • 开源拓展:开源特性使其易于扩展和定制。

  • 适用场景:当设备无GPU资源,却需运行大语言模型时,Llama.cpp是理想之选,让普通设备也能体验大语言模型的强大 。


四、SGLang 引擎:高效推理的潜力新星

开发者:未知

核心特色:SGLang专注高效推理,可能运用稀疏计算、分布式优化等技术提升性能,虽充满神秘感,但潜力无限。

显著优势

  • 场景优化:针对特定场景深度优化,显著提升推理效率。

  • 企业适配:适合对高性能推理有需求的企业级应用。

  • 适用场景:在大规模分布式环境中运行大语言模型,SGLang值得一试,是探索未来推理技术的重要窗口 。


五、MLX 引擎:高效计算的未来之光

开发者:未知

核心特色:MLX可能是针对大语言模型优化的机器学习框架,聚焦高效计算与推理,是高效计算领域的“未来之星”。

显著优势

  • 硬件适配:可能针对TPU或定制芯片等特定硬件进行优化。

  • 效率优先:适用于追求极致计算效率的场景。

  • 适用场景:若需在特定硬件上运行大语言模型,MLX值得关注,其潜在的硬件优化能力,有望引领未来高效计算 。


六、Ollama:本地大模型运行的便捷之选

开发者:社区项目

核心特色:Ollama是本地运行大语言模型的利器,支持LLaMA、GPT等多种模型,简化模型部署与运行流程。

显著优势

  • 简单易用:操作简便,适合个人用户与开发者。

  • 本地运行:无需云端资源,完全在本地设备实现模型运行。

  • 模型丰富:支持多种模型,使用灵活。

  • 适用场景:若想在个人设备上测试或运行大语言模型,Ollama是绝佳选择,助你摆脱云端依赖,随时体验大模型魅力


七、指标对比

1. 性能对比

引擎
性能特点
硬件支持
适用模型规模
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">Transformers
通用性强,性能中等,适合中小规模模型推理和训练。
CPU/GPU
中小规模模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(247, 247, 247);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">vLLM
高性能推理,通过 PagedAttention 等技术优化 GPU 内存和计算效率。
GPU
大规模模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">Llama.cpp
针对 CPU 优化,性能中等,适合资源受限的环境。
CPU
中小规模模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(247, 247, 247);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">SGLang
可能通过稀疏计算或分布式优化提升性能,具体性能取决于实现。
未知(可能 GPU)
中大规模模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">MLX
可能针对特定硬件(如 TPU 或定制芯片)优化,性能潜力高。
特定硬件
中大规模模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(247, 247, 247);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">Ollama
性能中等,适合本地运行,无需高性能硬件。
CPU/GPU
中小规模模型

总结:

    ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;" class="list-paddingleft-1">
  • vLLM在 GPU 上的推理性能最优,适合大规模模型。

  • Llama.cppOllama适合在 CPU 或低配设备上运行中小规模模型。

  • SGLangMLX的性能潜力较大,但需要更多实践验证。

ingFang SC", "Microsoft YaHei", SimHei, Arial, SimSun;font-size: 16px;color: rgb(79, 79, 79);line-height: 26px;font-synthesis-style: auto;overflow-wrap: break-word;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">2.并发能力对比

引擎
并发支持
适用场景
Transformers
支持多线程和多 GPU 推理,但并发能力受限于框架和硬件。
中小规模并发任务
vLLM
高并发支持,通过内存优化和批处理技术显著提升并发性能。
高并发推理任务
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">Llama.cpp
并发能力有限,适合低并发场景。
单任务或低并发任务
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", sans-serif;font-size: 14px;font-style: normal;font-variant-ligatures: no-common-ligatures;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: center;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(247, 247, 247);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">SGLang
可能通过分布式计算支持高并发,具体能力取决于实现。
中高并发任务
MLX
可能针对高并发优化,具体能力取决于硬件和实现。
中高并发任务
Ollama
并发能力中等,适合本地低并发任务。
单任务或低并发任务

总结:

  • vLLM在高并发场景下表现最佳,适合生产环境。

  • TransformersSGLang适合中等并发任务。

  • Llama.cppOllama更适合单任务或低并发场景。

3.适用场景对比

引擎
适用场景
优势
Transformers
研究、开发、中小规模生产环境。
功能全面,社区支持强大,适合多种 NLP 任务。
vLLM
大规模模型推理、高并发生产环境。
极致性能,高效内存管理,适合企业级应用。
Llama.cpp
资源受限环境(如嵌入式设备、低配服务器)。
轻量级,无需 GPU,适合低成本部署。
SGLang
中大规模模型推理、分布式计算环境。
可能通过优化提升性能,适合探索性项目。
MLX
特定硬件环境(如 TPU 或定制芯片)。
可能针对硬件优化,适合高性能计算场景。
Ollama
本地开发、测试、个人使用。
简单易用,无需云端资源,适合个人用户。

总结:

  • Transformers是通用性最强的工具,适合大多数 NLP 任务。

  • vLLM是企业级高并发场景的首选。

  • Llama.cppOllama适合个人开发者或资源受限的环境。

  • SGLangMLX适合需要高性能或特定硬件支持的场景。

4.硬件兼容性对比

引擎
硬件支持
适用设备类型
Transformers
CPU/GPU
普通服务器、个人电脑、云服务器
vLLM
GPU
高性能 GPU 服务器
Llama.cpp
CPU
低配设备、嵌入式设备
SGLang
未知(可能 GPU)
高性能服务器
MLX
特定硬件
TPU、定制芯片等
Ollama
CPU/GPU
个人电脑、普通服务器

总结:

  • TransformersOllama兼容性最强,支持多种设备。

  • vLLMSGLang需要高性能 GPU 或服务器。

  • Llama.cpp适合低配设备,而MLX需要特定硬件支持。

八、每秒输出token对比


1. 性能影响因素

在对比 TPS 之前,需要明确影响性能的关键因素:

  • 硬件性能: GPU 的算力、显存带宽、显存容量等。

  • 模型规模: 参数量越大,推理速度越慢。

  • 批处理大小(Batch Size): 较大的批处理可以提高吞吐量,但会增加显存占用。

  • 引擎优化: 不同引擎在内存管理、计算优化等方面的表现差异显著。


2.GPU 性能对比

以下是 A800、A100 和 H100 的主要参数对比:

GPU 型号
FP32 算力 (TFLOPS)
显存容量 (GB)
显存带宽 (TB/s)
适用场景
A800
19.5
40/80
2.0
推理、训练
A100
19.5
40/80
2.0
高性能计算、AI 训练
H100
30.0
80
3.35
高性能推理、AI 训练


  • H100是目前性能最强的 GPU,适合高吞吐量和高并发场景。

  • A100A800性能接近,但 A800 主要针对中国市场,符合出口管制要求。


3.引擎 TPS 对比

以下是各引擎在不同 GPU 上的预估 TPS(以 LLaMA-13B 模型为例):

引擎
A800 (TPS)
A100 (TPS)
H100 (TPS)
备注
Transformers
50-100
60-120
80-150
性能中等,适合中小规模推理。
vLLM
200-400
300-600
500-1000
高性能推理,优化显存和批处理。

说明:

  • vLLM 在高性能 GPU(如 H100)上的表现最佳,TPS 可达 500-1000,远超其他引擎。

  • Transformers 性能中等,适合通用场景。

  • Llama.cpp 和 Ollama 性能较低,适合资源受限的环境。

  • SGLang 和 MLX 的性能数据较少,需进一步测试。




九、这里简单介绍一下 Xinference 安装


Xinference 在 Linux, Windows, MacOS 上都可以通过pip来安装。如果需要使用 Xinference 进行模型推理,可以根据不同的模型指定不同的引擎。

如果你希望能够推理所有支持的模型,可以用以下命令安装所有需要的依赖:

pipinstall"xinference[all]"

备注

如果你想使用 GGML 格式的模型,建议根据当前使用的硬件手动安装所需要的依赖,以充分利用硬件的加速能力。更多细节可以参考Llama.cpp 引擎这一章节。

如果你只想安装必要的依赖,接下来是如何操作的详细步骤。

Transformers 引擎

PyTorch(transformers) 引擎支持几乎有所的最新模型,这是 Pytorch 模型默认使用的引擎:

pipinstall"xinference[transformers]"

vLLM 引擎

vLLM 是一个支持高并发的高性能大模型推理引擎。当满足以下条件时,Xinference 会自动选择 vllm 作为引擎来达到更高的吞吐量:

  • 模型格式为pytorchgptq或者awq

  • 当模型格式为pytorch时,量化选项需为none

  • 当模型格式为awq时,量化选项需为Int4

  • 当模型格式为gptq时,量化选项需为Int3Int4或者Int8

  • 操作系统为 Linux 并且至少有一个支持 CUDA 的设备

  • 自定义模型的model_family字段和内置模型的model_name字段在 vLLM 的支持列表中。

目前,支持的模型包括:

  • llama-2,llama-3,llama-2-chat,llama-3-instruct

  • baichuan,baichuan-chat,baichuan-2-chat

  • internlm-16k,internlm-chat-7b,internlm-chat-8k,internlm-chat-20b

  • mistral-v0.1,mistral-instruct-v0.1,mistral-instruct-v0.2,mistral-instruct-v0.3

  • codestral-v0.1

  • Yi,Yi-1.5,Yi-chat,Yi-1.5-chat,Yi-1.5-chat-16k

  • code-llama,code-llama-python,code-llama-instruct

  • DeepSeek,deepseek-coder,deepseek-chat,deepseek-coder-instruct

  • codeqwen1.5,codeqwen1.5-chat

  • vicuna-v1.3,vicuna-v1.5

  • internlm2-chat

  • qwen-chat

  • mixtral-instruct-v0.1,mixtral-8x22B-instruct-v0.1

  • chatglm3,chatglm3-32k,chatglm3-128k

  • glm4-chat,glm4-chat-1m

  • qwen1.5-chat,qwen1.5-moe-chat

  • qwen2-instruct,qwen2-moe-instruct

  • gemma-it

  • orion-chat,orion-chat-rag

  • c4ai-command-r-v01

安装 xinference 和 vLLM:

pipinstall"xinference[vllm]"

Llama.cpp 引擎

Xinference 通过llama-cpp-python支持ggufggml格式的模型。建议根据当前使用的硬件手动安装依赖,从而获得最佳的加速效果。

初始步骤:

pipinstallxinference

不同硬件的安装方式:

  • Apple M系列

    CMAKE_ARGS="-DLLAMA_METAL=on"pipinstallllama-cpp-python
  • 英伟达显卡:

    CMAKE_ARGS="-DLLAMA_CUBLAS=on"pipinstallllama-cpp-python
  • AMD 显卡:

    CMAKE_ARGS="-DLLAMA_HIPBLAS=on"pipinstallllama-cpp-python

SGLang 引擎

SGLang 具有基于 RadixAttention 的高性能推理运行时。它通过在多个调用之间自动重用KV缓存,显著加速了复杂 LLM 程序的执行。它还支持其他常见推理技术,如连续批处理和张量并行处理。

初始步骤:

pipinstall'xinference[sglang]'

十、Xinference环境变量

XINFERENCE_ENDPOINT

Xinference 的服务地址,用来与 Xinference 连接。默认地址是 http://127.0.0.1:9997,可以在日志中获得这个地址。

XINFERENCE_MODEL_SRC

配置模型下载仓库。默认下载源是 “huggingface”,也可以设置为 “modelscope” 作为下载源。

XINFERENCE_HOME

Xinference 默认使用<HOME>/.xinference作为默认目录来存储模型以及日志等必要的文件。其中<HOME>是当前用户的主目录。可以通过配置这个环境变量来修改默认目录。

XINFERENCE_HEALTH_CHECK_ATTEMPTS

Xinference 启动时健康检查的次数,如果超过这个次数还未成功,启动会报错,默认值为 3。

XINFERENCE_HEALTH_CHECK_INTERVAL

Xinference 启动时健康检查的时间间隔,如果超过这个时间还未成功,启动会报错,默认值为 3。

XINFERENCE_DISABLE_HEALTH_CHECK

在满足条件时,Xinference 会自动汇报worker健康状况,设置改环境变量为 1可以禁用健康检查。

XINFERENCE_DISABLE_VLLM

在满足条件时,Xinference 会自动使用 vLLM 作为推理引擎提供推理效率,设置改环境变量为 1可以禁用 vLLM。

XINFERENCE_DISABLE_METRICS

Xinference 会默认在 supervisor 和 worker 上启用 metrics exporter。设置环境变量为 1可以在 supervisor 上禁用 /metrics 端点,并在 worker 上禁用 HTTP 服务(仅提供 /metrics 端点)

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

链载AI是专业的生成式人工智能教程平台。提供Stable Diffusion、Midjourney AI绘画教程,Suno AI音乐生成指南,以及Runway、Pika等AI视频制作与动画生成实战案例。从提示词编写到参数调整,手把手助您从入门到精通。
  • 官方手机版

  • 微信公众号

  • 商务合作

  • Powered by Discuz! X3.5 | Copyright © 2025-2025. | 链载Ai
  • 桂ICP备2024021734号 | 营业执照 | |广西笔趣文化传媒有限公司|| QQ