Problem with Xformers wheel from jetson-ai-lab

[Ncouch64]

Hello, I have pulled the most recent wheels available on jetson-ai-lab for the Jetson Orin NX (jp6/cu126) and have verified torch, torchaudio, torchvision, torch-trt, and onnxgpu are all working as expected. However, after pulling and installing xformers:

`Python 3.10.12 (main, Aug 15 2025, 14:32:43) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.

import xformers
WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for:
PyTorch 2.8.0 with CUDA 1209 (you have 2.8.0)
Python 3.12.3 (you have 3.10.12)
Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers)
Memory-efficient attention, SwiGLU, sparse and more won't be available.
Set XFORMERS_MORE_DETAILS=1 for more details
`

I have also attempted to build xformers from source, but it will overwrite the pre-built pytorch install with the most recent version of torch which does not work.

Am I perhaps missing something?

[Ncouch64]

Perhaps I will need to build PyTorch from source.