AutoGPTQ

一键量化大模型,显存减半推理加速

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 547 🐛 待处理 272 👀 关注 3 ⚠️ 作者已归档(停止维护)
GitHub ↗
GitHub 星标 ★ 5069
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,无付费版本。

项目预览

AutoGPTQ 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队AutoGPTQ
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型deep-learning,inference,large-language-models,llms,nlp,pytorch,quantization,transformer,transformers
GitHub 星标★ 5069
上线时间2023-04-13 00:00:00
代码更新2025-04-11
仓库体积8.2 MB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数11

项目介绍

AutoGPTQ 是一个基于 GPTQ 算法的 LLM 量化工具包,提供简洁易用的 API,帮助开发者将大语言模型压缩为低比特(如 4-bit)表示,从而大幅降低显存占用和推理成本。它解决了大模型在消费级硬件上部署困难的问题,支持 Hugging Face Transformers 生态,可一键量化主流模型,并提供了量化后模型的推理加速。核心能力包括:多种量化配置(如 4-bit、8-bit)、支持 CPU/GPU 推理、与 transformers 无缝集成、以及高效的量化内核。通过 AutoGPTQ,开发者无需深入底层算法即可完成模型量化,适合在资源受限环境中部署大模型。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 5069

使用教程

难度:进阶 约 15 分钟 部署方式:库/依赖 7 步

环境要求

  • 操作系统:仅支持 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. 1确认平台与GPU

    确认系统为 Linux 或 Windows,并核对 CUDA/ROCm 版本与显卡架构(NVIDIA 需高于 Maxwell),再选择对应安装命令。

  2. 2pip安装(CUDA 12.1)

    CUDA 12.1 环境直接安装官方预编译 wheel,无需额外索引地址,务必带 --no-build-isolation。

    pip install auto-gptq --no-build-isolation
  3. 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/
  4. 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. 5可选Triton后端

    如需 Triton 后端可加装 [triton] 依赖,目前仅支持 Linux,且不支持 3-bit 量化。

    pip install auto-gptq[triton] --no-build-isolation
  6. 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. 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···

★ 215 2026-10-17
AgentGym 开源

让大模型智能体在多样环境中自我进化,一键复现论文实验

Code and implementations for the ACL 2025 paper "AgentGym: Evolving Large Language M···

★ 857 2026-08-09
tirex 开源

零样本预测长短周期,无需训练直接上手

TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context···

★ 308 2026-08-09
SoftCoT 开源

用软思维链替代文本推理,大幅降低 LLM 推理成本

ACL'2025: SoftCoT: Soft Chain-of-Thought for Efficient Reasoning with LLMs. and prep···

★ 96 2026-08-09
CAME 开源

减少大模型训练显存,优化器也能自适应省内存

[ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory O···

★ 100 2026-08-09
model-optimization 开源

几行代码压缩模型,让大模型跑进手机和嵌入式设备

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including q···

★ 1582 2026-10-17
LECO 开源

用LoRA轻量擦除扩散模型中的指定概念,不动原模型

Low-rank adaptation for Erasing COncepts from diffusion models.

★ 323 2026-08-09
LLM-Drop 开源

丢弃冗余层,加速 Transformer 推理,省算力不掉点。

The official implementation of the paper "Uncovering the Redundancy in Transformers ···

★ 193 2026-08-09
EasySteer 开源

不重训模型,推理时插拔式控制 LLM 行为

[EMNLP 2026] A Unified Framework for High-Performance and Extensible LLM Steering

★ 313 2026-10-07