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


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.


Have you install the framework with CUDA supported?
For example, below is the link to install PyTorch with CUDA support:


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 "

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

please correct me if I think wrong?

Thanks in advance,

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.


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