Can I execute yolov5 on the GPU of JETSON AGX XAVIER?

Autonomous Machines Jetson & Embedded Systems Jetson AGX Xavier
8

Hi,

Really sorry for the late reply.

Since Jetson is an ARM platform, you will need the PyTorch and TorchVision that are built with ARM architecture.
For JetPack 5.0, you can find PyTorch v1.11.0 and v1.12.0 in the below topic:

We also have a container that has PyTorch pre-installed.

For running YOLOv5 on Jetson, it’s recommended to try our Deepstream SDK.
It has optimized the pipeline based on the Jetson hardware and is expected to give you a better performance.
Below is a sample from the community for your reference:

Thanks.