I can't use gpu on jetson orin nano

[yasinkorus]

I use Jetson Orin Nano.
python3 -c “import torch; print(torch.cuda.is_available())” is true
print(“CUDA is available” if torch.cuda.is_available() else “CUDA is not available”) is 12.2
python3 -c “import torch; print(torch.version)” is 2.3.0a0+6ddf5cf85e.nv24.04
I can’t use GPU
when I write this code python3 setup.py install --user, I am getting error.
How ı can fix? Thanks you for answer

[TomNVIDIA]

Hello,

Thanks for visiting the NVIDIA Developer forums! Your topic will be best served in the Jetson category.

Jetson & Embedded Systems

I will move this post over for visibility.

Cheers,
Tom

[AastaLLL]

Hi,

In your environment, PyTorch has built with CUDA support but TorchVision doesn’t.
We also have a prebuilt TorchVision package with GPU enabled for the JetPack 6 users.

Please find the below topic for more info:

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers. PyTorch pip wheels JetPack 6

PyTorch v2.3.0

JetPack 6.0 (L4T R36.2…

Thanks.

[yasinkorus]

thanks