How to use Cuda with torch in jetson orin nano with jetpack 5.1.2

Autonomous Machines Jetson & Embedded Systems Jetson Orin Nano
1

I have jetson orin nano with 8 gb ram and jetpack 5.1.2 where it showing open cv with cuda is yes but when import torch and run for cuda it display false . I ha d python version 3.8.10.So,how i can use cuda or GPU for this.

3

Hi,

These are different libraries so please check which one (or both?) you need for the CUDA support.

PyTorch:
We provide the prebuilt with CUDA support.
Please find the installation instructions below:

OpenCV:
The default package in JetPack doesn’t build with CUDA support.
Please build it from the source. You can find a script in the below link:

Thanks.

5

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