项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-10 · 当前星标 ★ 3388
使用教程
环境要求
- Python 3.7 及以上版本
- PyTorch 1.4 及以上版本
- 已安装可用的 pip
- 操作系统建议为 Ubuntu 16.04/18.04/20.04、macOS 10.15、Windows 10 或 WSL(官方测试环境)
安装与启动步骤
-
1确认环境版本
先确认本机 Python 版本满足 3.7+,并已安装 PyTorch 1.4+,Catalyst 依赖 PyTorch 运行。
python --version -
2安装 Catalyst
使用 pip 从 PyPI 安装最新版 Catalyst,-U 表示升级到最新版本。
pip install -U catalyst -
3按需装扩展模块
做机器学习相关任务装 catalyst[ml],做计算机视觉任务装 catalyst[cv],会额外拉取依赖。
pip install catalyst[ml] -
4安装 master 版本
需要仓库最新开发代码时,直接从 GitHub master 分支安装并升级。
pip install git+https://github.com/catalyst-team/catalyst@master --upgrade -
5验证安装
导入 catalyst 查看版本号,无报错说明安装成功,可继续跑 README 中的示例脚本。
python -c "import catalyst; print(catalyst.__version__)"
如何确认成功
执行 python -c "import catalyst" 无报错即安装成功,随后可运行 README 的 MNIST 示例训练脚本。
常见问题
Q:安装 Catalyst 需要什么运行环境?
A:需要 Python 3.7+ 和 PyTorch 1.4+,官方在 Ubuntu 16.04/18.04/20.04、macOS 10.15、Windows 10 与 WSL 上测试通过。
Q:需要额外功能依赖怎么办?
A:按任务类型安装扩展包:pip install catalyst[ml] 安装机器学习相关依赖,pip install catalyst[cv] 安装计算机视觉相关依赖。
Q:如何安装最新的开发版本?
A:执行 pip install git+https://github.com/catalyst-team/catalyst@master --upgrade,直接从 GitHub 的 master 分支安装。
Q:安装后怎么快速上手?
A:参照 README 的 Getting started 示例,用 catalyst.dl 的 SupervisedRunner 配合 PyTorch 的 DataLoader 编写训练循环。
注意事项
- README 节选中的训练示例代码被截断,实际使用时请以仓库 README 的完整示例为准。
- 建议在虚拟环境(venv/conda)中安装,避免与已有 PyTorch 环境产生依赖冲突。
- catalyst[ml] 与 catalyst[cv] 会额外安装依赖,按实际任务需要选择其一或都不装。
核心亮点
- 高度模块化的 Runner 和 Callback 系统,可灵活定制训练流程,减少样板代码
- 内置丰富 CV 任务(分类/分割/检测)的预配置模型和指标,开箱即用
- 无缝支持混合精度、多 GPU/TPU 分布式训练,提升实验效率
不足之处
- 文档和社区活跃度相比 PyTorch Lightning 等竞品稍弱,新手上手成本略高
- 项目更新节奏放缓,部分功能依赖较旧版本的 PyTorch
适用场景
- 学术研究中的快速实验原型和对比验证
- 工业级 CV 模型的训练与部署流水线搭建
- 需要统一管理多个实验配置和指标的团队协作
替代项目
PyTorch Lightning、Ignite、fastai
同类项目推荐
DS2
开源
用 LLM 评分自动筛数据,少数据也能训出好模型
[ICLR 2025] Official implementation of paper "Improving Data Efficiency via Curating···
STEPQuant
开源
定位循环状态量化误差,让低比特 LLM 更稳
STEPQuant: When and Where Errors Matter in Delta-Rule Recurrent State Quantization
zenbytes
开源
跟着 Notebook 一步步上手 ZenML,搞定 MLOps 流水线
A simple guide to MLOps through ZenML and its various integrations.
LightRFT
开源
用奖励模型驱动微调,让 LLM/VLM 按目标越训越强
LightRFT: Light, Efficient, Omni-modal & Reward-model Driven Reinforcement Fine-Tuni···
torchebm
开源
不跑采样链,GPU 上直接训 EBM 和流匹配
Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives f···
regress-lm
开源
把回归变成生成,用预训练模型直接预测数值
Library for sequence-to-sequence numeric prediction, applicable to any tokenizable i···
gpt-neox
开源
在GPU集群上训练GPT级大模型的开源并行框架
An implementation of model parallel autoregressive transformers on GPUs, based on th···
finetuned-qlora-falcon7b-medical
开源
用QLoRA微调Falcon-7B,打造心理健康对话助手
Finetuning of Falcon-7B LLM using QLoRA on Mental Health Conversational Dataset
Kardia-R1
开源
让AI学会共情,用强化学习提升情感支持回复质量
[WWW] Kardia-R1: Unleashing LLMs to Reason toward Understanding and Empathy for Emot···
