安装

关于支持的平台/操作系统和硬件,请查看 Github 上的兼容性 (Compatibility)部分。

二进制分发版

你可以通过 pip 安装最新稳定版的 Triton

pip install triton

CPython 3.10-3.14 提供二进制 wheel 包。

从源码安装

Python 包

你可以通过运行以下命令从源码安装 Python 包

git clone https://github.com/triton-lang/triton.git
cd triton

pip install -r python/requirements.txt # build-time dependencies
pip install -e .

请注意,如果你的系统中没有安装 llvm,setup.py 脚本将会下载官方的 LLVM 静态库并进行链接。

如需使用自定义 LLVM 进行构建,请查看 Github 上的使用自定义 LLVM 构建 (Building with a custom LLVM)部分。

然后,你可以通过运行测试来验证安装是否成功

# One-time setup
make dev-install

# To run all tests (requires a GPU)
make test

# Or, to run tests without a GPU
make test-nogpu