Baike.dev
All toolsAI codingTrendingOpen sourceNewsSubmit
Log in
Back to tool/Back to issues
#1946·jetson-inference

Problem with Xformers wheel from jetson-ai-lab

Author: Ncouch64Created Nov 6, 2025Updated Nov 6, 2025

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

View original on GitHubView discussion on GitHub