Expecting torch and torchvision for JP6.1

Support YOLO
, , ,
2

I would recommend checking the Jetson Dockerfile in the Ultralytics repo for straightforward setup. AFAIK, this should work for any Jetpack version, but to be honest, I’ve never set up a Jetson device personally, you’ll likely need to modify this line in the Dockerfile:

- FROM nvcr.io/nvidia/l4t-jetpack:r36.3.0
+ FROM nvcr.io/nvidia/l4t-jetpack:r36.4.0

Since r36.4.0 has Jetpack 6.1 according to this NVIDIA page

Otherwise, see the NVIDIA Jetson Download Center for the latest PyTorch version 2.5.0 but doesn’t look like they have a wheel for torchvision (yet).

1 Like