PyTorch and Torchvision for Jetpack 5.1.2

Hi, I’m using a Jetson AGX Orin with Jetpack 5.1.2

I see that I need to install PyTorch and Torchvision as described here PyTorch for Jetson if I want to have CUDA access. However I do not see which PyTorch version and corresponding Torchvision version to use for Jetpack 5.1.2. The only Torch version that appears to be supported by Jetpack 5.1.2 is Pytorch 2.1.0, but that does not have a corresponding Torchvision version. I also do not see version correspondences on the Pytorch website.

So how can I set up Torch and Torchvision on a Jetson with Jetpack 5.1.2?

Hi @alsozatch, for PyTorch 2.1, I’d use torchvision v0.16 or v0.15. The PyTorch 2.0 wheels should work on JetPack 5.1.2, but while you’re at it, you my as well pick up the latest. You could also use these pre-built torchvision or PyTorch container images if you prefer:

Ok thanks, PyTorch 2.0.0 and Torchvision 0.15.1 seem to be working correctly with CUDA on Jetpack 5.1.2

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