Yolov5 docker container cannot detect GPU/CUDA on Jetson Orin

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

Hello everyone,

I have a problem with my yolov5 docker container. I have load my yolov5 container from [Preformatted text](https://hub.docker.com/r/ultralytics/yolov5). I have researched about the similar topic and problem, which were caused by torch and torchvision version. However, I have update torch to version 1.13 and torchvision to version 0.14 for my container.

I have check that my Jetson Orin got CUDA and its version is 11.4. I have also run jetson-inference container with CUDA without any problems and the container could detect CUDA on Orin.

Thank you so much

3

Hi,

Not sure if the container has been built with GPU/CUDA support.

Could you give our NGC container a try?

Thanks.

5

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