PyTorch

Check your PyTorch version: Triton 3.2 works with torch >= 2.6 . I recommend to upgrade to torch 2.6 because there are several improvements to torch.compile.

Triton 3.1 works with torch >= 2.4 . torch 2.3.x and older versions are not supported.

Remember to install PyTorch with CUDA 12.

CUDA

CUDA 12 is required. CUDA 11.x and older versions are not supported. The wheels here are built against CUDA 12.6, and they should work with other CUDA 12.x.

Choose either of the following ways to install CUDA: