Problem with installing torchvision on Jetson Orin Nano

Hello everyone,

I have a problem with the installation of torchvision on the Jetson Orin Nano Development Kit. I recently upgraded the Jetpack version from 6.0 to 6.1. I also already installed the latest PyTorch with the installation guides (Installing PyTorch for Jetson Platform - NVIDIA Docs, PyTorch for Jetson Platform - NVIDIA Docs). However, when I now try to install torchvision by first cloning the github repository and then running the setup.py it says ModuleNotFoundError: No module named ‘torch’, although when testing if torch is installed, it confirms that it is indeed installed. Now I don’t know how to install the fitting torchvision version. This procedure orginally worked when I installed torchvision for the Jetpack 6.0 version before. I hope someone can help me with this problem.

Thanks in advance.

Hi,

All the JetPack 6.1 DL related packages can be found below:

http://jetson.webredirect.org/jp6/cu126

Would you mind giving it a try?
Thanks.

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