Hi @jose.donate, have you installed one of the PyTorch wheels for JetPack 6 from this topic? These were built with CUDA enabled:
Given that you said your torch version is 2.1.1, and there isn’t a torch+CUDA wheel for that specific version, my guess is that you just did pip3 install torch, and the aarch64 wheels on PyPi are not built with CUDA support.