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 大模型的逐层加载推理方案

3月前 AI开源项目 798 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 语音平台,能听能说能聊
推荐阅读
  • 阿里开源zg:把ripgrep+BM25+向量检索整合为Agent时代本地统一检索层
  • TranslateGemma:Google开源专业翻译模型,55语言覆盖+图文识别
  • figures4papers:耶鲁博士开源的论文配图脚本库,8000+ Star 一键生成出版级图表
  • Skill Seekers|自动将文档转化为Claude技能的开源工具
  • 15套免费数据大屏模板开源项目:HTML原生构建,覆盖财务/电商/物流等主流行业
评论 (0)
请登录后发表评论
分类精选
WeKnora:终于等到了腾讯ima的开源知识库框架,用 API 轻松打造本地智能文档检索
13295 1年前
OpenSpec:比 Cursor Plan 更聪明?试试这款让 AI 编码更靠谱的规范驱动工具
11200 1年前
CapCut API:一个剪映API开源项目,让AI自动剪辑视频
11145 10月前
AIRI:你的开源AI女友,让你随时拥有属于自己的 AI VTuber
10098 1年前
Antigravity-Manager:这个开源神器让你白嫖ClaudeOpus 4.5,Gemini 3!还能接Claude Code等任意平台
9790 10月前
就要创作:从提示词到创作团队,开源 AI 网文写作平台
9417 1年前
CompressO:开源免费的视频压缩神器,让你的硬盘瞬间轻松 10 倍
9289 1年前
AiNiee:开源AI翻译工具,完美解决游戏、小说、字幕、文档的批量翻译
8694 1年前
awesome-openclaw-skills:700+ Skills 一条命令装配完成,如何让本地 AI Agent 真正落地可用
8680 9月前
Claude Code 生成专业图表的 15 个 Skills:覆盖 7 种渲染引擎的完整指南
8658 6月前

文章目录

关于苏米客
分类排行
1 anidoodle:用代码作画的开源工具,32种风格插画动画一键生成
2 Quiet.md:不到20MB的本地富文本笔记,完美保留飞书文档结构
3 agent-desktop 开源:Vercel Labs 打造桌面端 AI Agent,后台运行不抢鼠标
4 global-law:用 Git 版本管理全球法律,精确到字的变更追溯
5 Claude Code + 6 个开源 Skill:用代码生成视频,告别 AI 视频抽卡时代
6 9 个开源视频动画项目:Agent 驱动的代码绘图、手绘视频与卡通短片制作
7 Codex 用量条:让侧边栏直观显示剩余额度的开源插件
8 android-ad-cleaner:AI 整顿完电视后,现在轮到手机了,ADB 自动关广告
9 Vercel Labs 开源 agent-desktop:基于无障碍访问树的桌面 AI Agent 操作工具
10 DeepSeek 开源 Super-agent 系统:AI 自主管理多任务,跨平台持续在线工作
苏米客
登录 注册
显示
没有账号? · 立即注册
显示
已有账号? · 直接登录
登录代表你已同意 用户协议 和 隐私政策
©2015-2024 苏米客XMSUMI 版权所有 · WWW.XMSUMI.COM 闽ICP备14005900号-6 网站地图 百度地图
拉美跨境导航 微信文章助手 程序库 免费影视APP 免费字体下载 产品经理导航 爱克硕儿 产品经理AI资讯 Axure元件库下载 申请友联