Yolov5: RuntimeError: CUDA error: out of memory

[LokedSher]

❔Question

Hey guys, I am a green hand and it is my first time to run yolov5. I read README and did exactly as the 'tutorial-Train custom data' tells and I encountered this problem. I don't known is my GPU occupied too much or something else is wrong ?

Additional context

[github-actions[bot]]

Hello @LokedSher, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook , Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

• Cloud-based AI systems operating on hundreds of HD video streams in realtime.
• Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
• Custom data training, hyperparameter evolution, and model exportation to any destination.

For more information please visit https://www.ultralytics.com.

[NanoCode012]

@LokedSher , It means that you don’t have enough available GPU memory.

Also, check out the Multi-GPU Training on ReadMe to properly utilize multi-gpus.

[Frank1126lin]

Yeah，just use small batch-size to aviod Mem problem.

[meizano]

Your GPU doesn't have enough memory. Change your batch size to lower number by giving parameter such as: --batch 8

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.