10+年产品经理专注分享AI 工具、AI 资讯、AI Coding、Vibe Coding与下一代产品创新,按 Ctrl+D 收藏我们
关于我 留言板 小程序 标签云

苏米客

  • 首页
  • AIGC
    • AI最新动态
    • AI学习教程
    • AI工具集合
    • AI产品百科
    • AI编程开发
    • AI提示词
    • AI开源项目
    • AI智能体
  • Axure
    • Axure动态
    • Axure教程
  • 产品
    • 用户体验
    • 产品设计
    • 苏米杂谈
  • 资源
    • 产品UI组件库
    • 开源图标库
    • 中后台框架
  • 书单
    • AI书籍
    • 用户体验
    • UI视觉
    • 产品研究
    • 其他类型
  • 下载
    • Axure组件
    • Axure原型
    • 文档报告
    • 素材资源
  • 登录
  • 首页
  • AIGC
    • AI最新动态
    • AI学习教程
    • AI工具集合
    • AI产品百科
    • AI编程开发
    • AI提示词
    • AI开源项目
    • AI智能体
  • Axure
    • Axure动态
    • Axure教程
  • 产品
    • 用户体验
    • 产品设计
    • 苏米杂谈
  • 资源
    • 产品UI组件库
    • 开源图标库
    • 中后台框架
  • 书单
    • AI书籍
    • 用户体验
    • UI视觉
    • 产品研究
    • 其他类型
  • 下载
    • Axure组件
    • Axure原型
    • 文档报告
    • 素材资源
当前位置: 首页 » AI开源项目

AirLLM:4GB 显存跑 70B 大模型的逐层加载推理方案

1小时前 AI开源项目 12 0

之前通过 Ollama、SGLang 在本地部署过大模型的朋友都知道,受限于显卡显存,基本上只能部署小参数且量化后的模型。

这些模型在本地跑基本是残血状态,拿来玩玩还可以,但效果往往差强人意。

最近在 GitHub 上发现一个 3 万 Star 的项目,名字很直白:AirLLM。它的标语就一句话——70B inference with single 4GB GPU。

70B 全精度权重一百多 G,4GB 显存连个零头都不够,凭什么能跑?下面我们一起来看看。

AirLLM 逐层加载原理

核心原理:逐层加载

推理一个 Transformer,常规做法是把模型所有层都放进显存,算完一层进入下一层。

AirLLM 的思路完全不同——它一次只在 GPU 上留一层,算完这层,把下一层从磁盘或内存读进来,再把刚才那层扔出去。这么滚动着跑,整个模型永远只有「薄薄一片」在显卡上。

所以显存占用不是「模型总大小」,而是「单层大小 + 一点缓存」。70B 的单层也就几百 M,4GB 显卡绰绰有余。

预取机制示意图

还有个配套优化叫预取(prefetching):上一层在算的时候,它提前把下一层从磁盘读出来,让「加载」和「计算」重叠,别让显卡干等。

4bit 量化压缩

此外还支持可选的 4bit 块量化压缩(只压权重部分,不是整个模型量化),官方称能再快 3 倍,精度损失几乎可忽略。

关键点:AirLLM 主打「不用量化、不用蒸馏、不用剪枝」就能跑大模型。这话的意思不是它反对量化,而是哪怕一个 bit 都不压,原模原样,它也能靠逐层搬运把模型跑起来。保的是精度,牺牲的是速度。

实际能跑多大

官方给了一张「推理时显存占用」的参考表:

模型规模 推理显存占用
8B 级别 1-2GB
235B 级别 约 3GB
70B 级别 约 4GB
405B(Llama 3.1) 约 8GB
671B(DeepSeek-V3) 约 12GB
2.8T(Kimi K3,最大开源 MoE) 低于 4GB

AirLLM 显存占用参考表

看到 Kimi K3 那行很值得注意:2.8 万亿参数的模型,单卡 4GB 以下就能跑。当然,这是「能跑起来出 token」,不是「跑得快」。

MoE 模型之所以更省,是因为它是稀疏的,逐专家流式加载,不需要把整个专家群都搬进显存。

怎么使用

AirLLM 是个 Python 库,一行命令安装:

pip install airllm

调用方式和 transformers 几乎一样,换个 AutoModel 就行:

from airllm import AutoModel

model = AutoModel.from_pretrained("Qwen/Qwen3-32B")

input_text = ['What is the capital of United States?']
input_tokens = model.tokenizer(
    input_text, return_tensors="pt",
    return_attention_mask=False, truncation=True,
    max_length=128, padding=False
)
output = model.generate(
    input_tokens['input_ids'].cuda(),
    max_new_tokens=20, use_cache=True,
    return_dict_in_generate=True
)
print(model.tokenizer.decode(output.sequences[0]))

想再快点,传个 compression='4bit' 就行。

Mac 上也能跑(Apple Silicon + MLX),甚至纯 CPU 也能推理。支持几乎所有主流开源 LLM:Llama、Qwen、DeepSeek、Mistral、Phi、Gemma、ChatGLM……

总结

AirLLM 干的事情其实很「搞机」:它没有让模型变小,而是让硬件的边界变模糊了。

你不用买 A100、不用依赖云端 API、不用忍受量化掉精度,只要有一张消费级显卡或者一台 Mac,就能原模原样地把 70B 甚至更大的模型跑起来。代价是慢、需要自己写代码、得忍受磁盘 IO 瓶颈。

如果你手头有张小显卡,又馋大模型的能耐,值得一试。

GitHub 仓库:https://github.com/lyogavin/airllm

声明:本站原创文章文字版权归本站所有,转载务必注明作者和出处;本站转载文章仅仅代表原作者观点,不代表本站立场,图文版权归原作者所有。如有侵权,请联系我们删除。
未经允许不得转载:AirLLM:4GB 显存跑 70B 大模型的逐层加载推理方案
#AirLLM #大模型推理 #显存优化 #本地部署 #开源工具 
收藏 1
MiniMax-H3 视频提示词写作教程:T2VA/I2VA/FL2VA/L2VA 四种任务全解析
阿里 CosyVoice Studio 实测:一站式 AI 语音平台,能听能说能聊
推荐阅读
  • EdgeClaw:断网也能跑的小龙虾,OpenClaw 的安全增强版
  • MinerU:一款全能的PDF文档解析神器,本地部署的PDF智能解析方案
  • BiliSum 开源工具:B 站、YouTube 视频自动转写,生成结构化笔记和思维导图
  • n-skills:为中小团队量身定做的AI技能模块化框架,一个真正可用的开源技能集
  • Kami 开源文档设计系统:统一设计规范,让 AI 生成的文档直接可用
评论 (0)
请登录后发表评论
分类精选
WeKnora:终于等到了腾讯ima的开源知识库框架,用 API 轻松打造本地智能文档检索
11479 11月前
OpenSpec:比 Cursor Plan 更聪明?试试这款让 AI 编码更靠谱的规范驱动工具
10372 10月前
CapCut API:一个剪映API开源项目,让AI自动剪辑视频
9204 7月前
Antigravity-Manager:这个开源神器让你白嫖ClaudeOpus 4.5,Gemini 3!还能接Claude Code等任意平台
8513 7月前
AIRI:你的开源AI女友,让你随时拥有属于自己的 AI VTuber
8219 11月前
CompressO:开源免费的视频压缩神器,让你的硬盘瞬间轻松 10 倍
7884 10月前
就要创作:从提示词到创作团队,开源 AI 网文写作平台
7862 10月前
awesome-openclaw-skills:700+ Skills 一条命令装配完成,如何让本地 AI Agent 真正落地可用
7647 6月前
baoyu-skills:又一个宝藏Skill,面向内容创作者的技能集,支持图文生成、发布与处理
7143 6月前
Claude Code 生成专业图表的 15 个 Skills:覆盖 7 种渲染引擎的完整指南
6962 3月前

文章目录

关注「苏米客」公众号

订阅推送更及时,手机查看更方便
分类排行
1 Prime Agent:会自我沉淀经验的开源 AI 代理架构,9.2k Stars
2 AirLLM:4GB 显存跑 70B 大模型的逐层加载推理方案
3 Firecrawl anydoc 开源:14 种文档格式转 Markdown,比 LibreOffice 快 256 倍
4 SkillsGate 开源项目:AI Agent 技能的统一管理器,91000+ 技能一键安装到 18+ 款工具
5 腾讯 WorkBuddy Bench 开源:260 道 Agent 评测任务 + 2 个自动化 Skill
6 微软开源 Skill Recorder:录制操作自动生成 AI Agent SKILL.md
7 AnyDoc:Rust 文档转 Markdown 引擎,性能碾压 MarkItDown
8 scroll-world 与 img2threejs:两个炫酷的 GitHub 开源项目,让 AI 帮你做网页和 3D 模型
9 Humanizer-zh:专治中文写作 AI 味的开源 Skill,14.6k Star
10 SenseNova U1.5-Lite-Preview 开源:8B 轻量级统一多模态模型,支持 4K 图像生成与编辑
©2015-2024 苏米客XMSUMI 版权所有 · WWW.XMSUMI.COM 闽ICP备14005900号-6
微信文章助手 程序库 免费影视APP 免费字体下载 产品经理导航 爱克硕儿 产品经理AI资讯 Axure元件库下载 申请友联