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

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


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


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

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