Installation¶
Binary Distributions¶
You can install the latest stable release of Triton from pip:
pip install triton
Binary wheels are available for CPython 3.6-3.9 and PyPy 3.6-3.7.
And the latest nightly release:
pip install -U --pre triton
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