Pytorch and torchvision

[akajaykumar349]

Can someone help with the latest version installation of pytorch and torchvision.

[AastaLLL]

Hi,

You can find the package and instructions in the below topic.
We also have some docker container that has PyTorch installed.

PyTorch for Jetson - version 1.11 now available

Below are pre-built PyTorch pip wheel installers for Python on Jetson Nano, Jetson TX1/TX2, Jetson Xavier NX/AGX, and Jetson AGX Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). PyTorch pip wheels

PyTorch v1.12.0

JetPack 5.0 Developer Preview (L4T R34.1.0) / J…

Thanks.

[akajaykumar349]

Ok Thank you, I will look into this.

[magnus.gabell]

Hi,

i have a topic on this too. You did not mention how to install torchvision for Pytorch 1.12 and Jetpack 5.0.1 DP?
I want to make it work for AGX Orin and your pip wheel for PyTorch help me get those packages in but crash on torchvision. Seems like Pytorch 1.12 and Torchvision 0.11 is not compatible.

[kayccc]

Hi magnus.gabell,

Please help to open a new topic for your issue. Thanks