大模型部署Ollama和vLLM:Ollama是什么?vLLM是什么? 分页注意力机制是什么?
Ollama是一款专注于简化大型语言模型本地部署和运行的开源框架,基于Go语言实现,支持跨平台运行,并以“开箱即用”为核心理念,适合个人开发者和轻量化场景。而vLLM是一个高效的大模型推理与服务引擎,基于PyTorch构建,创新性地引入了PagedAttention技术,旨在解决大模型服务中的显存效率与吞吐量瓶颈,支持张量并行和流水线并行,可横向扩展至多机多卡集群。
Ollama是一款专注于简化大型语言模型本地部署和运行的开源框架,基于Go语言实现,支持跨平台运行,并以“开箱即用”为核心理念,适合个人开发者和轻量化场景。
而vLLM是一个高效的大模型推理与服务引擎,基于PyTorch构建,创新性地引入了PagedAttention技术,旨在解决大模型服务中的显存效率与吞吐量瓶颈,支持张量并行和流水线并行,可横向扩展至多机多卡集群。
一、Ollama
Ollama是什么?
Ollama 是一个专注于本地运行大型语言模型(LLM)的工具,支持macOS/Linux/Windows,ARM架构设备友好,提供简洁的命令行接口,适合个人开发者和研究者快速体验模型。
Ollama以“开箱即用”为核心理念,专为个人开发者和轻量化场景设计。
Ollama基于Go语言实现,通过模块化封装将模型权重、依赖库和运行环境整合为统一容器。这种设计使得用户无需关注底层依赖,仅需一条命令行即可启动模型服务。
Ollama的优势在于开发友好性,但生产部署时面临性能、稳定性和运维能力的全方位挑战。对于关键业务系统,建议仅将其作为实验阶段的验证环节,而非最终部署方案。
# Linux/macOS 一键安装脚本
curl -fsSL https://ollama.com/install.sh | sh
# 拉取 DeepSeek-R1 模型
ollama pull DeepSeek-R1
# 启动 DeepSeek-R1 模型
ollama run DeepSeek-R1 # 启动交互式对话
# 通过 API 调用 DeepSeek-R1 模型
curl http://localhost:11434/api/generate -d '{
"model": "DeepSeek-R1",
"prompt": "为什么天空是蓝色的?"
}'
二、vLLM
vLLM(Vectorized Large Language Model Serving System)是什么?
vLLM 是一个高效的大模型推理与服务引擎,旨在解决大模型服务中的显存效率与吞吐量瓶颈,适合生产环境部署。
# 安装 vLLM
pip install vllm # 需要 Python 3.8+ 和 CUDA 11.8+
#启动 vLLM 推理服务,并使用 DeepSeek-R1 模型
# 单卡启动 DeepSeek-R1
python -m vLLM.entrypoints.api_server \
--model deepseek/DeepSeek-R1 \
--tensor-parallel-size 1
# 使用 curl 命令调用 DeepSeek-R1 模型的推理服务
curl http://localhost:8000/generate \
-H "Content-Type: application/json" \
-d '{"prompt": "解释量子纠缠", "max_tokens": 200}'
# 使用 vLLM 的 Python SDK 调用 DeepSeek-R1 模型
from vllm import LLM
llm = LLM("deepseek/DeepSeek-R1")
outputs = llm.generate(["AI 的未来发展方向是"])
print(outputs)
分页注意力机制(PagedAttention)是什么?
分页注意力机制借鉴了计算机操作系统中的内存分页管理,通过动态分配和复用显存空间,显著提升大模型推理的效率和吞吐量。

在传统的大模型推理中,注意力机制(Transformer的自注意力层)需要为每个请求的序列分配连续的显存块,存储以下数据:
(1)键值缓存(Key-Value Cache,KV Cache):存储历史token的键值对,用于生成后续token。
(2)中间激活值:计算注意力权重时的中间结果。
vLLM基于PyTorch构建,创新性地引入了PagedAttention技术。该技术借鉴操作系统的虚拟内存分页机制,将注意力键值对(KV Cache)存储在非连续显存空间,显著提高了显存利用率。
PagedAttention通过分块管理显存、动态按需分配和跨请求共享内存,解决了传统方法中显存碎片化、预留浪费和并发限制三大瓶颈。

三、如何系统学习掌握AI大模型?
AI大模型作为人工智能领域的重要技术突破,正成为推动各行各业创新和转型的关键力量。抓住AI大模型的风口,掌握AI大模型的知识和技能将变得越来越重要。
学习AI大模型是一个系统的过程,需要从基础开始,逐步深入到更高级的技术。
这里给大家精心整理了一份
全面的AI大模型学习资源,包括:AI大模型全套学习路线图(从入门到实战)、精品AI大模型学习书籍手册、视频教程、实战学习、面试题等,资料免费分享!

1. 成长路线图&学习规划
要学习一门新的技术,作为新手一定要先学习成长路线图,方向不对,努力白费。
这里,我们为新手和想要进一步提升的专业人士准备了一份详细的学习成长路线图和规划。可以说是最科学最系统的学习成长路线。

2. 大模型经典PDF书籍
书籍和学习文档资料是学习大模型过程中必不可少的,我们精选了一系列深入探讨大模型技术的书籍和学习文档,它们由领域内的顶尖专家撰写,内容全面、深入、详尽,为你学习大模型提供坚实的理论基础。(书籍含电子版PDF)

3. 大模型视频教程
对于很多自学或者没有基础的同学来说,书籍这些纯文字类的学习教材会觉得比较晦涩难以理解,因此,我们提供了丰富的大模型视频教程,以动态、形象的方式展示技术概念,帮助你更快、更轻松地掌握核心知识。

4. 2024行业报告
行业分析主要包括对不同行业的现状、趋势、问题、机会等进行系统地调研和评估,以了解哪些行业更适合引入大模型的技术和应用,以及在哪些方面可以发挥大模型的优势。

5. 大模型项目实战
学以致用 ,当你的理论知识积累到一定程度,就需要通过项目实战,在实际操作中检验和巩固你所学到的知识,同时为你找工作和职业发展打下坚实的基础。

6. 大模型面试题
面试不仅是技术的较量,更需要充分的准备。
在你已经掌握了大模型技术之后,就需要开始准备面试,我们将提供精心整理的大模型面试题库,涵盖当前面试中可能遇到的各种技术问题,让你在面试中游刃有余。

全套的AI大模型学习资源已经整理打包,有需要的小伙伴可以
微信扫描下方CSDN官方认证二维码,免费领取【保证100%免费】

火山引擎开发者社区是火山引擎打造的AI技术生态平台,聚焦Agent与大模型开发,提供豆包系列模型(图像/视频/视觉)、智能分析与会话工具,并配套评测集、动手实验室及行业案例库。社区通过技术沙龙、挑战赛等活动促进开发者成长,新用户可领50万Tokens权益,助力构建智能应用。
更多推荐
所有评论(0)