安装¶
关于支持的平台/操作系统和支持的硬件,请查阅 Github 上的兼容性部分。
二进制发行版¶
您可以通过 pip 安装 Triton 的最新稳定版本
pip install triton
二进制 wheel 文件适用于 CPython 3.10-3.14。
从源码安装¶
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 构建部分。
然后,您可以通过运行测试来检验您的安装
# 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