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?
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.