Files
triton/_sources/getting-started/installation.rst.txt

54 lines
1.1 KiB
Plaintext
Raw Normal View History

2021-03-05 22:49:39 -05:00
==============
2021-03-06 17:35:11 -05:00
Installation
2021-03-05 22:49:39 -05:00
==============
2021-03-23 17:10:07 -04:00
---------------------
Binary Distributions
---------------------
2021-07-28 09:06:14 +00:00
You can install the latest stable release of Triton from pip:
2021-07-28 10:15:45 +00:00
.. code-block:: bash
2021-07-28 09:06:14 +00:00
pip install triton
Binary wheels are available for CPython 3.6-3.9 and PyPy 3.6-3.7.
And the latest nightly release:
2021-03-23 17:10:07 -04:00
.. code-block:: bash
pip install -U --pre triton
2021-03-06 17:35:11 -05:00
--------------
From Source
--------------
2021-03-05 22:49:39 -05:00
+++++++++++++++
Python Package
+++++++++++++++
You can install the Python package from source by running the following commands:
.. code-block:: bash
2021-07-31 05:27:59 +00:00
git clone https://github.com/openai/triton.git;
2021-03-05 22:49:39 -05:00
cd triton/python;
2021-07-28 09:06:14 +00:00
pip install cmake; # build time dependency
2021-03-05 22:49:39 -05:00
pip install -e .
2021-07-31 05:27:59 +00:00
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.
2021-03-23 17:10:07 -04:00
2021-03-05 22:49:39 -05:00
You can then test your installation by running the unit tests:
.. code-block:: bash
pytest -vs .
and the benchmarks
.. code-block:: bash
cd bench/
2021-07-28 09:06:14 +00:00
python -m run --with-plots --result-dir /tmp/triton-bench