Installation¶
Binary Distributions¶
You can install the latest stable release of Triton from pip:
pip install triton
Binary wheels are available for CPython 3.7-3.11 and PyPy 3.8-3.9.
And the latest nightly release:
pip install -U --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ triton-nightly
From Source¶
Python Package¶
You can install the Python package from source by running the following commands:
git clone https://github.com/openai/triton.git;
cd triton/python;
pip install cmake; # build-time dependency
pip install -e .
Note that, if llvm-11 is not present on your system, the setup.py script will download the official LLVM11 static libraries link against that.
You can then test your installation by running the unit tests:
pip install -e '.[tests]'
pytest -vs test/unit/
and the benchmarks
cd bench
python -m run --with-plots --result-dir /tmp/triton-bench