Problem with GPU when training YoloV4 on workstation
1 vue (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Hi everyone,
I am doing the video detection task on Matlab with YoloV4. I have followed all the stesps as the instructions on MATLAB (https://au.mathworks.com/help/vision/ug/object-detection-using-yolov4-deep-learning.html).
To increase the speed of training, i have change the training option to GPU with "ExcutionEnvironment". However, I am facing the problem with the gpu same like this:
Can anyone recommend the solution for me? Thank you so much.
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1106155/image.png)
My training dataset inclde 527 images and the training options and gpu properties are:
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1106160/image.png)
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1106165/image.png)
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1106170/image.png)
1 commentaire
Joss Knight
le 1 Sep 2022
It looks like a memory issue. You have a 32GiB card but only 110MB available. I suspect some basic underlying features can't work with that little memory. Trying resetting your device (reset(gpuDevice)) then put a breakpoint before the call to trainYOLOv4Detector and check that you have plenty of memory available. If that's okay we may have to check the size of your input data.
Réponses (0)
Voir également
Catégories
En savoir plus sur Image Data Workflows dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!