No pytorch for jetson orin nano Jetpack 6.2.1

AI & Data Science NGC Catalog Feedback
, , ,
1

According to the installation instructions

pytorch should be installed from Index of /compute/redist/jp by selecting the corresponding Jetpack version (612) but this is not available.

What is the current method for installing pytorch on Orin Nano with the latest supported Jetpack?

2

Hi @clynamen,

Hope you are doing well!

You can try installing Pytorch with the following command:

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126

Best regards,
Nico

3

Thanks @nico.morua !

This makes pytorch available, but I wonder if this was compiled for the Orin Nano:


    self.alpha_mask = torch.ones_like(resized_image_rgb[0:1, ...].to(self.data_device))

torch.AcceleratorError: CUDA error: no kernel image is available for execution on the device

Search for `cudaErrorNoKernelImageForDevice' in 
 for more information.

Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
4

Hi @clynamen ,

I also found this forum which may help you installing PyTorch: Torch on Jetson Jetpack 6.2 - #3 by SivaRamaKrishnaNV

Hope this helps!

Best regards,
Nico

5

thanks, what worked eventually was to

  1. remove all installed pytorch versions (with sudo too. )
  2. install pytorch from https://pypi.jetson-ai-lab.io
  3. 
# note: .io replaces the old .dev link
sudo pip install https://pypi.jetson-ai-lab.io/jp6/cu126/+f/62a/1beee9f2f1470/torch-2.8.0-cp310-cp310-linux_aarch64.whl#sha256=62a1beee9f2f147076a974d2942c90060c12771c94740830327cae705b2595fc
  4. Install the cuda libs as suggested here
  5. sudo apt install --reinstall libcusolver-12-6 libcusolver-dev-12-6
6

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.