No wheels for arm

[surak]

Describe the bug

There are only x86_64 wheels, no aarch64, and only on pip, not here.

Environment details

Triton 3.1, gh200

[22dimensions]

+1

[SulRash]

You can build it from source:

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

You also need to install pytorch through by explicitly selecting a wheel if you're installing for grace hopper first:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

Afterwards run the tests to check if everything works:

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

My tests all pass, using a GH200

[surak]

I know I can and that’s not the point, it is? But thanks for the recipe.

[atalman]

Hi @surak You can install latest arm wheel using this command:

pip3 install triton --index-url https://download.pytorch.org/whl/test/cu126 --force-reinstall