Problem with GPU when training YoloV4 on workstation

1 次查看（过去 30 天）

Canh Long Nguyen 2022-8-23

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1784455-problem-with-gpu-when-training-yolov4-on-workstation

评论： Joss Knight 2022-9-1

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.

My training dataset inclde 527 images and the training options and gpu properties are:

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Joss Knight 2022-9-1

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.

请先登录，再进行评论。

请先登录，再回答此问题。