Install PyTorch on Jetson Orin Nano

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
1

I have Jetson Orin Ubuntu 22.04 JetPack 6.2, CUDA 12.6.68, cuDNN 9.3.0.75, and I can’t install PyTorch. Please Help. Thanks.

jetson@ubuntu:~$ python3
Python 3.10.12 (main, Feb 4 2025, 14:57:36) [GCC 11.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import torch
Traceback (most recent call last):
File “”, line 1, in
File “/home/jetson/.local/lib/python3.10/site-packages/torch/init.py”, line 237, in
from torch._C import * # noqa: F403
ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory

4

Hi,

Please try below commands:

wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/6cc/6ecfe8a5994fd/torch-2.6.0-cp310-cp310-linux_aarch64.whl#sha256=6cc6ecfe8a5994fd6d58fb6d6eb73ff2437428bb4953f3ebaa409f83a5f4db99
pip install torch-2.6.0-cp310-cp310-linux_aarch64.whl
wget  https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/aa2/2da8dcf4c4c8d/torchvision-0.21.0-cp310-cp310-linux_aarch64.whl#sha256=aa22da8dcf4c4c8dc897e7922b1ef25cb0fe350e1a358168be87a854ad114531
pip install torchvision-0.21.0-cp310-cp310-linux_aarch64.whl
2 Likes
5

Thanks: import torch
print(torch.cuda.is_available())
True

1 Like
7

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