Install PyTorch on Jetson Orin Nano

Robotics & Edge Computing Jetson Systems Jetson Orin Nano
,
1

Hello! I need to install GPU-accelerated PyTorch on my Jetson Orin Nano ( JetPack = 6.2.1 ) for the lerobot project. But the lerobot requires torch<2.8.0,>=2.2.1, and torchvision<0.23.0,>=0.21.0. It seems that only the combinations of torch=2.7.0 with torchvision=0.22.0 or torch=2.6.0 with torchvision=0.21.0 can meet my requirements. Could you please provide the installation packages for these corresponding versions? Thanks.

6

Hi,

Since we don’t have prebuilt wheels, please try to build it from source using below commands

  1. clone jetson-containers
git clone https://github.com/dusty-nv/jetson-containers.git
  1. follow the jetson-containers/docs/setup.md at master · dusty-nv/jetson-containers · GitHub
  2. build from source
./build.sh pytorch:2.7-builder torchvision:0.22.0-builder

Thanks