ktransformers

单卡甚至纯CPU跑大模型,显存不够也能玩转MoE推理微调

KTransformers 是一个面向大语言模型推理与微调的开源异构优化框架,核心目标是让开发者在单台消费级或服务器级机器上,以极低成本体验和部署大模型。它通过 CPU/GPU 混合计算、MoE 稀疏激活、量化与算子融合等技术,把原本需要多卡 A100 才能跑动的模型(如 DeepSeek、Qwen 等大参数 MoE 模型)压缩到单卡甚至纯 CPU 环境运行。框架支持推理加速和 LoRA 微调两条路径,提供与 HuggingFace 生态兼容的接口,用户无需重写模型代码即可接入。项目由清华大学团队主导,社区活跃,在中文开发者圈有较高知名度。它解决的核心问题是显存不足与硬件门槛过高,让个人研究者和小团队也能本地跑大模型、做实验和二次开发。

开源 free 模型训练
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 19524
维护状态 活跃
是否开源
定价模式 free

项目数据

分类模型训练
开发团队kvcache-ai
所属国家
定价模式free
价格说明开源免费,采用Apache-2.0许可证
访问状态
是否开源
开源协议Apache-2.0
主要语言Python
技术栈/模型
GitHub 星标★ 19524
30天Star增速
HF 下载量
上线时间2024-07-26 00:00:00
最近更新2026-09-18 00:00:00
维护状态活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-09-16
浏览次数0

使用教程

难度:高级 约 30 分钟 部署方式:本地安装 6 步

⚠ 该项目 README 未提供完整安装说明,以下步骤为通用流程,请以官方文档为准。

环境要求

  • Python 环境(项目开发语言为 Python)
  • 支持 CPU-GPU 异构计算的机器(消费级或服务器级)
  • 能访问 GitHub 以克隆仓库
  • 需自行阅读推理与微调子文档获取完整安装步骤

安装与启动步骤

  1. 1克隆项目仓库

    从 GitHub 获取源码,README 未给出安装命令,克隆后需自行查阅子文档。

    git clone https://github.com/kvcache-ai/ktransformers.git
  2. 2进入项目目录

    切换到克隆下来的仓库根目录,后续文档路径均以此为基准。

    cd ktransformers
  3. 3阅读推理文档

    README 指向 kt-kernel/README.md 作为推理能力的入口说明,先看它再动手。

    cat kt-kernel/README.md
  4. 4阅读微调手册

    SFT 能力见 KTransformers-Fine-Tuning_Cookbook.md,含硬件检查、安装与配方。

    cat doc/en/SFT/KTransformers-Fine-Tuning_Cookbook.md
  5. 5按手册完成环境检查

    先按手册中的硬件检查环节确认本机 CPU、内存与显存能否承载目标模型。

  6. 6选择推理或微调路线

    推理走 kt-kernel 路径,SFT 走 LlamaFactory 路线,二者安装步骤不同须分开处理。

如何确认成功

README 未提供启动命令,按所引文档执行后,以对应子文档给出的验证方式为准。

常见问题

Q:README 里为什么没有 pip install 之类的安装命令?

A:本次节选只给出总览与文档入口,具体安装步骤在 kt-kernel/README.md 和 SFT Cookbook 里,需点进子文档查看。

Q:没有独立显卡能跑吗?

A:项目定位是通过 CPU-GPU 异构计算、MoE 稀疏激活与量化降低门槛,纯 CPU 或单卡场景的支持情况要以子文档为准。

Q:支持哪些模型?

A:README 更新记录提到 DeepSeek、Qwen 等大参数 MoE 模型,以及 GLM-5.3-flash、Kimi K2.5/K2.6 的 LoRA 微调。

Q:微调需要准备什么?

A:微调手册覆盖硬件检查、安装、BF16/FP8/INT8 配方、LoRA 与全量微调、资源规划和排错,建议先通读。

注意事项

  • 本次 README 节选未包含任何可直接执行的安装命令,切勿凭猜测执行包名或参数。
  • 推理与微调是两条独立路径,安装依赖不同,不要混用同一套步骤。
  • 微调手册建议按顺序阅读:硬件检查 → 安装 → 精度配方 → 资源规划 → 排错。
  • 项目更新较快,模型支持列表请以仓库最新更新记录为准。

核心亮点

  • 支持CPU+GPU异构混合推理,显著降低显存门槛,单卡可跑大参数MoE模型
  • 针对MoE稀疏激活做专门优化,推理速度相比朴素实现有明显提升
  • 兼容HuggingFace模型与LoRA微调流程,接入成本低,无需重写代码

不足之处

  • 配置与依赖较复杂,不同硬件组合下需手动调参,新手入门曲线陡
  • 主要优化集中在特定模型架构(如MoE),通用性覆盖有限

适用场景

  • 个人开发者用单张消费级显卡本地部署大参数MoE模型做推理测试
  • 小团队在有限GPU资源下对开源大模型进行LoRA微调实验
  • 研究者对比异构推理优化方案,验证CPU/GPU混合计算性能

替代项目

llama.cpp、vLLM

项目介绍

ktransformers 是一个模型训练领域的开源项目,官方简介:A Flexible Framework for Experiencing Heterogeneous LLM Inference/Fine-tune Optimizations。项目使用 Python 开发,在 GitHub 上获得 19517 星标。

上一篇:SteptronOss

下一篇:notebooks

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 61327 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1279 2026-08-13