Liger-Kernel

环境要求

• 已安装 Python 与 pip 环境
• 已安装 PyTorch（ROCm 环境需先装 ROCm 版 PyTorch）
• 具备可访问 GitHub 与 PyPI 的网络环境
• 需要 Triton 支持的 GPU 环境（用于运行融合内核）

安装与启动步骤

1. 1克隆项目仓库

   从 GitHub 拉取 Liger-Kernel 源码，后续所有安装命令都在仓库根目录执行。

   git clone https://github.com/linkedin/Liger-Kernel.git
   cd Liger-Kernel

   复制

2. 2安装默认依赖

   README 推荐的可编辑安装方式，setup.py 会自动检测本地后端并选择默认依赖。

   pip install -e .

   复制

3. 3装开发依赖(可选)

   需要参与开发或跑测试时使用，安装 dev 附加依赖组。

   pip install -e ".[dev]"

   复制

4. 4ROCm 环境准备

   在 ROCm 上需先从 PyTorch ROCm 索引安装 ROCm 版 PyTorch，再执行上面的安装。

   pip3 install torch torchvision --index-url https://download.pytorch.org/whl/rocm7.2

   复制

5. 5安装 cuTile 依赖

   需要 cuTile 支持时安装该可选项；也可换用带 tileiras 编译器的版本。

   pip install -e ".[cutile]"

   复制

6. 6安装 CuTe DSL

   需要 NVIDIA CUTLASS Python DSL 支持时安装该附加依赖组。

   pip install -e ".[cutedsl]"

   复制

如何确认成功

在 Python 中执行 import liger_kernel 不报错，即表示包已成功安装到当前环境。

常见问题

注意事项

• README 未给出端口、模型路径等运行时配置项，本教程仅覆盖安装环节。
• 命令请务必在克隆得到的 Liger-Kernel 仓库根目录下执行，否则 -e . 会找不到 setup.py。
• 高级用法与 API 细节请查阅官网 https://linkedin.github.io/Liger-Kernel 。
• 遇到问题可通过 README 中公布的 Discord 频道 https://discord.gg/X4MaxPgA 反馈。