How to run customized YOLO V5 model on Nvidia Orin GPU cores

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Orin
1

Hi,

we downloaded basic Yolo v5 model from GitHub repo & r trying to run on GPU cores,
while running/selecting as device id using GPU, getting some errors
GPU not found, so how to run it on GPU?
how to run the YOLO-V5 model on GPU?

Please let me know the procedure to run the model on GPU…

Thanks in Advance,
L V GiriBabu.

3

Hi,

Have you install the framework with CUDA supported?
For example, below is the link to install PyTorch with CUDA support:
https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html

Thanks.

4

Thanks for ur reply,

Yes, our current framework supported the CUDA .

even we installed torch & torch vision packages also.

but while selecting device ID to GPU, model is throwing error " GPU not found "

image
image693×39 2.39 KB

we are thinking like GPU core will detect/active after install Pytorch with CUDA.

please correct me if I think wrong?

Thanks in advance,

5

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.
Thanks

Hi,

How do you install the PyTorch?
Could you share the instructions with us?

Thanks.