Problem with Xformers wheel from jetson-ai-lab
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?
Source: dusty-nv/jetson-inference