Torch and torchvision cuda 12.6 jetpack 6.1

Robotics & Edge Computing Jetson Systems Jetson Nano
1

What is the procedure for installing torch and torchvision in jetson nano orin with jetpack 6.1 and cuda 12.6?
I tried multiple ways to install different versions, none seem to work.
I also tried : PyTorch for Jetson
I couldnt get it work.
Any help is appreciated.

Thanks

4

Hi,

Please refer to the Jetson/L4T/TRT Customized Example - eLinux.org
to get JP6.1 compatible torch and torchvision.

Thanks

5

Thank you so much for the reference.
I did follow this one and it worked, however, I faced some numpy compatibility issue. I had to downgrade the one installed with this torch and torchvision from v2.x to v1.x

6

Hi,

Downgrade the numpy to v1.x is expected behavior since it have some dependence issue.

Thanks

8

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