How to set up torch with GPU support on Jetson Orin (with Miniconda)?

Autonomous Machines Jetson & Embedded Systems Jetson Orin Nano
1

Hi,

How to setup torch with GPU support on Jetson Orin?
Please guide.

Thanks

2

Hi @ayhmdalilabzns - you can find the PyTorch pip wheels that were built with CUDA enabled for Jetson here:

There are also containers like l4t-pytorch and various ones in jetson-containers that you can use with torchvision, transformers, ect and lots of other packages built on top of PyTorch using GPU.

5

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