
AutoGPTQ
一键量化大模型,显存减半推理加速
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 5069
使用教程
环境要求
- 操作系统:仅支持 Linux 和 Windows
- NVIDIA 显卡需高于 Maxwell 架构(Maxwell 及更低不支持)
- CUDA 11.8 / CUDA 12.1 或 ROCm 5.7 环境,且需匹配对应的 PyTorch(如 2.2.1+cu118、2.2.1+cu121、2.
- 已安装 Python 与 pip;从源码安装还需能编译 PyTorch 扩展
安装与启动步骤
-
1确认平台与GPU
确认系统为 Linux 或 Windows,并核对 CUDA/ROCm 版本与显卡架构(NVIDIA 需高于 Maxwell),再选择对应安装命令。
-
2pip安装(CUDA 12.1)
CUDA 12.1 环境直接安装官方预编译 wheel,无需额外索引地址,务必带 --no-build-isolation。
pip install auto-gptq --no-build-isolation -
3pip安装(CUDA 11.8)
CUDA 11.8 环境需通过 autogptq-index 的 cu118 索引安装预编译包,否则会尝试本地编译。
pip install auto-gptq --no-build-isolation --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/ -
4pip安装(ROCm 5.7)
AMD ROCm 5.7 环境使用 rocm573 索引安装对应的预编译 wheel。
pip install auto-gptq --no-build-isolation --extra-index-url https://huggingface.github.io/autogptq-index/whl/rocm573/ -
5可选Triton后端
如需 Triton 后端可加装 [triton] 依赖,目前仅支持 Linux,且不支持 3-bit 量化。
pip install auto-gptq[triton] --no-build-isolation -
6从源码安装
克隆仓库后先装 numpy、gekko、pandas 等构建依赖,再以 -e 方式本地安装;不建议关闭 CUDA 扩展编译。
git clone https://github.com/PanQiWei/AutoGPTQ.git && cd AutoGPTQ pip install numpy gekko pandas pip install -vvv --no-build-isolation -e . -
7运行测试验证
在仓库根目录执行测试用例,确认安装与量化内核可用;需先完成源码克隆并进入项目目录。
pytest tests/ -s
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
use_marlin | 否 | 加载模型时启用 Marlin int4*fp16 矩阵乘内核加速 | use_marlin=True |
BUILD_CUDA_EXT | 否 | 设为 0 可跳过 PyTorch 扩展编译,但会退化为慢速 Python 实现 | BUILD_CUDA_EXT=0 |
如何确认成功
运行 pytest tests/ -s,测试通过且无编译或导入报错即表示安装成功。
常见问题
Q:安装命令执行失败怎么办?
A:先确保已安装 numpy gekko pandas 等构建依赖;若仍失败,可按 README 提示尝试最后的兜底方式 python setup.py install。
Q:为什么我的 NVIDIA 显卡不被支持?
A:AutoGPTQ 在 NVIDIA 系统上不支持 Maxwell 或更低架构的 GPU,需要更高级别的显卡。
Q:Triton 后端有什么限制?
A:Triton 后端目前仅支持 Linux 系统,并且不支持 3-bit 量化。
Q:Marlin 内核需要什么硬件?
A:仅在计算能力 8.0 或 8.6 的设备(Ampere GPU)上可用,加载模型时传 use_marlin=True 启用。
Q:这个项目还在维护吗?
A:README 顶部已标注 AutoGPTQ 处于 unmaintained 状态,官方建议改用 GPTQModel 获取修复与新模型支持。
注意事项
- README 明确标注项目已不再维护,建议新项目改用 GPTQModel。
- 除非万不得已,不要设置 BUILD_CUDA_EXT=0,否则会回退到慢速的 Python 实现。
- 安装务必保留 --no-build-isolation 参数,否则可能编译失败。
- Transformers、optimum 和 peft 已集成 auto-gptq,可直接在生态内运行与训练 GPTQ 模型。
核心亮点
- API 简洁,几行代码完成模型量化,上手快
- 与 Hugging Face Transformers 深度集成,兼容主流模型
- 量化后推理速度快,显存占用显著降低,适合消费级显卡
不足之处
- 文档偏少,部分高级用法需读源码
- 对非 Transformer 架构模型支持有限
适用场景
- 在 8GB 显存显卡上部署 7B/13B 模型
- 批量压缩模型以降低云服务成本
- 为移动端或边缘设备准备轻量模型
替代项目
bitsandbytes、GPTQ-for-LLaMa、llama.cpp
同类项目推荐
nntrainer
开源
让手机家电自己训练模型,数据不出设备
NNtrainer is Software Framework for Training and Inferencing Neural Network Models o···
AgentGym
开源
让大模型智能体在多样环境中自我进化,一键复现论文实验
Code and implementations for the ACL 2025 paper "AgentGym: Evolving Large Language M···
tirex
开源
零样本预测长短周期,无需训练直接上手
TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context···
SoftCoT
开源
用软思维链替代文本推理,大幅降低 LLM 推理成本
ACL'2025: SoftCoT: Soft Chain-of-Thought for Efficient Reasoning with LLMs. and prep···
CAME
开源
减少大模型训练显存,优化器也能自适应省内存
[ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory O···
model-optimization
开源
几行代码压缩模型,让大模型跑进手机和嵌入式设备
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including q···
LECO
开源
用LoRA轻量擦除扩散模型中的指定概念,不动原模型
Low-rank adaptation for Erasing COncepts from diffusion models.
LLM-Drop
开源
丢弃冗余层,加速 Transformer 推理,省算力不掉点。
The official implementation of the paper "Uncovering the Redundancy in Transformers ···
EasySteer
开源
不重训模型,推理时插拔式控制 LLM 行为
[EMNLP 2026] A Unified Framework for High-Performance and Extensible LLM Steering