Unable to run YOLOv5 on Jetson Orin not use GPU and CUDA

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

I am using virtual environment on Jetson Orin but I have a problem with the YOLOv5 since torch not detect GPU, the application run on CPU and also when i made torch cuda is available() i got false

image

2

Hi @a.elhanashi, did you install one of the PyTorch wheels from here that was built with CUDA support?

https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html

Alternatively, you could use the l4t-pytorch container which comes with PyTorch/torchvision and CUDA pre-installed inside the container.

5

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