CTranslate2

让 Transformer 模型在 CPU/GPU 上跑得更快,省时省力

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 基础设施 🍴 复刻 555 🐛 待处理 270 👀 关注 53
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 4744
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

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

项目数据

分类基础设施
开发团队OpenNMT
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言C++
技术栈/模型avx,avx2,cpp,cuda,deep-learning,deep-neural-networks,gemm,inference,intrinsics,machine-translation,mkl,neon,neural-machine-translation,onednn,openmp,opennmt,parallel-computing,quantization,thrust,transformer-models
GitHub 星标★ 4744
上线时间2019-09-23 00:00:00
代码更新2026-11-09
仓库体积17.9 MB
本站数据更新2026-11-09
维护状态活跃
收录时间2026-08-10
浏览次数7

项目介绍

CTranslate2 是一个专为 Transformer 类模型设计的高性能推理引擎,用 C++ 编写,支持 CPU(含 AVX、AVX2、NEON 等指令集)和 GPU(CUDA)加速。它解决了深度学习模型在生产环境中部署时推理速度慢、内存占用高的问题,通过权重量化、层融合、内存复用等技术,在保持精度的前提下显著提升吞吐量并降低延迟。核心能力包括:支持多种 Transformer 架构(如 BERT、GPT、T5 等)、动态批处理、多 GPU 推理、与 Hugging Face Transformers 无缝集成,并提供了 Python 和 C++ API。项目在机器翻译、文本生成等场景中表现优异,尤其适合对延迟和成本敏感的工业级部署。

Star 增长趋势

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

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

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

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

环境要求

  • Python 环境及 pip(README 使用 pip 安装)
  • 已转换为 CTranslate2 优化格式的模型文件
  • 如需 GPU 加速:NVIDIA CUDA 环境;AMD ROCm 用户需下载 releases 页面的专用 wheel

安装与启动步骤

  1. 1安装 CTranslate2

    README 明确给出的唯一安装方式,通过 pip 安装 C++/Python 推理库,CPU 与 GPU 支持随包提供。

    pip install ctranslate2
  2. 2转换模型格式

    兼容模型必须先用官方转换器转为优化格式,README 只列出各框架指南链接,未给出具体命令,需按来源框架查阅文档。

  3. 3运行翻译推理

    用 Translator 加载转换后的翻译模型路径,再调用 translate_batch 传入分词后的 token 列表。

    import ctranslate2
    
    translator = ctranslate2.Translator(translation_model_path)
    translator.translate_batch(tokens)
  4. 4运行文本生成

    用 Generator 加载生成类模型(如 GPT、Llama 等)路径,再调用 generate_batch 传入起始 token。

    import ctranslate2
    
    generator = ctranslate2.Generator(generation_model_path)
    generator.generate_batch(start_tokens)

如何确认成功

执行 python -c "import ctranslate2; print(ctranslate2.__version__)",能打印出版本号即安装成功。

常见问题

Q:CTranslate2 支持哪些模型?

A:编码器-解码器(T5、BART、NLLB、Whisper 等)、仅解码器(GPT-2、Llama、Mistral、Qwen2 等)、仅编码器(BERT、DistilBERT、XLM-RoBERTa)。

Q:如何把已有模型转成 CTranslate2 格式?

A:库内置 OpenNMT-py、OpenNMT-tf、Fairseq、Marian、OPUS-MT、Transformers 的转换器,README 只给出官方文档链接,需按对应指南操作。

Q:可以用 AMD 显卡加速吗?

A:可以,README 说明针对 AMD ROCm GPU 在 GitHub releases 页面提供专用 Python wheel,需手动下载安装。

注意事项

  • 兼容模型必须先转换为 CTranslate2 优化格式才能推理,README 未给出转换命令,请查阅官方文档。
  • AMD ROCm GPU 用户不要直接用 pip 默认 wheel,应到 releases 页面下载对应版本。
  • 项目面向生产环境并提供向后兼容保证,但模型压缩与推理加速相关功能仍属实验性。

核心亮点

  • 推理速度比原生 PyTorch 快数倍,内存占用更低
  • 支持 INT8/FP16 量化,精度损失小
  • 与 Hugging Face 模型无缝集成,上手简单

不足之处

  • 仅支持 Transformer 架构,不覆盖 CNN/RNN 等模型
  • 文档和社区生态相对较小,高级用法需自行探索

适用场景

  • 生产环境中的机器翻译服务
  • 低延迟文本生成(如聊天机器人)
  • 资源受限的 CPU 服务器上部署大模型

替代项目

ONNX Runtime、TensorRT、OpenVINO

上一篇:inference

下一篇:casbin-gateway

同类项目推荐

MiMo2API 开源

把小米 MiMo 网页对话变成 OpenAI 兼容 API,现有工具直接调用

将小米 MiMo AI Studio 网页端对话转换为 OpenAI + Anthropic 兼容 API(Chat Completions···

★ 154 2026-10-12
Uncensored-Local-Studio 开源

零配置本地运行无审查 AI,图像、对话、语音一站式搞定

Portable local AI studio for Windows, Linux, and macOS. Zero-setup GUI for Image Gen···

★ 1977 2026-08-09
TensorRT-Model-Connect 开源

两条命令把 PyTorch 模型跑成 TensorRT 推理

From PyTorch model to end-to-end TensorRT inference experience in two commands—AI-n···

★ 291 2026-10-24
restai 开源

一键将多种AI模型封装为API,可视化搭建AI服务

RESTai is an AIaaS (AI as a Service) open-source platform. Supports many public and ···

★ 517 2026-08-09
magnitude 开源

按你的硬件自动调内核,本地跑模型更快

Open source inference engine that tunes its kernels on your hardware to find the fas···

★ 7766 2026-10-29
radar 开源

一个 Go 二进制搞定 K8s 可视化与 AI 运维诊断

The missing open-source Kubernetes UI with a built-in MCP server for AI agents. See ···

★ 4074 2026-08-09
toshllm 开源

让 Intel Mac 的 AMD 显卡也能本地跑大模型

Run large language models locally on Intel Macs with AMD GPUs - native macOS app wit···

★ 247 2026-09-15
VectorHub 开源

向量搜索入门资源库,已停更,新用户请转向 SIE 引擎

Deprecated historical repo. Superlinked now develops SIE, a self-hosted inference en···

★ 533 2026-08-09
LLaMA_MPS 开源

在 Mac 上让 LLaMA 跑得更快,用上 GPU 加速推理。

Run LLaMA (and Stanford-Alpaca) inference on Apple Silicon GPUs.

★ 575 2026-08-09