JetPack 6.1 on Jetson Orin Nano super

[758899275]

When I import torch, he gives an error:
from torch._C import * # noqa: F403
ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory
In the /usr/lib/aarch64-linux-gnu directory, my cudnn version is libcudnn.so libcudnn.so.9 libcudnn.so.9.3.0
torch is the latest version downloaded from PyTorch for Jetson

[DavidDDD]

Hi,

Do you install jetpack before import torch
For example:

sudo apt update
sudo apt install nvidia-jetpack

Thanks

[758899275]

I’ve already installed

download the torch from the software you provided
torch 2.6.0rc1
torchaudio 2.5.0
torchvision 0.20.0

[758899275]

I’ve fixed this issue, but there’s another problem
RuntimeError: operator torchvision::nms does not exist

[DavidDDD]

Hi

Please refer to this doc to install latest torch and torchvision which verified in JP6.1 .

Thanks

[758899275]

This is the latest version of torch and torchvision that I can find from the official, but it is still an error, can you provide me with the download link?
Thanks

[DavidDDD]

Hi,

Try below commands:

$ wget http://jetson.webredirect.org/jp6/cu126/+f/5cf/9ed17e35cb752/torch-2.5.0-cp310-cp310-linux_aarch64.whl#sha256=5cf9ed17e35cb7523812aeda9e7d6353c437048c5a6df1dc6617650333049092
$ pip install torch-2.5.0-cp310-cp310-linux_aarch64.whl
$ wget http://jetson.webredirect.org/jp6/cu126/+f/5f9/67f920de3953f/torchvision-0.20.0-cp310-cp310-linux_aarch64.whl#sha256=5f967f920de3953f2a39d95154b1feffd5ccc06b4589e51540dc070021a9adb9
$ pip install torchvision-0.20.0-cp310-cp310-linux_aarch64.whl 

Thanks

[758899275]

Thank you very much, this issue was successfully resolved