安装

有关支持的平台/操作系统和支持的硬件，请查看 Github 上的兼容性部分。

二进制分发版

您可以从 pip 安装 Triton 的最新稳定版本

pip install triton

二进制 wheel 包适用于 CPython 3.8-3.12 和 PyPy 3.8-3.9。

以及最新的 nightly 版本

pip install -U --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ triton-nightly

从源码构建

Python 包

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

git clone https://github.com/triton-lang/triton.git;
cd triton/python;
pip install ninja cmake wheel; # build-time dependencies
pip install -e .

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

有关使用自定义 LLVM 进行构建的信息，请查看 Github 上的使用自定义 LLVM 构建部分。

然后您可以通过运行单元测试来测试您的安装

pip install -e '.[tests]'
pytest -vs test/unit/

和基准测试

cd bench
python -m run --with-plots --result-dir /tmp/triton-bench