How can I use the function 'Predict' in deep learning toolbox with full GPU capacity

Question

Huhufly 2021-5-19

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/834468-how-can-i-use-the-function-predict-in-deep-learning-toolbox-with-full-gpu-capacity

评论： Huhufly 2021-5-25

I'm currently learning to use the deep learning toolbox to train a U-net for image regression tasks. When I was trying to use the 'predict' function with a minibatch size of 512 to generate some output, I found that the GPU memory usage was fluctuating periodically from 3.3/11GB to 9/11GB. I am pretty new to the matlab so I am wondering what is the cause of the phenomenon and how can I maximize the GPU useage so as to further accelerate the computation. Could anyone give me a little hint or suggestions?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Srivardhan Gadila 2021-5-22

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/834468-how-can-i-use-the-function-predict-in-deep-learning-toolbox-with-full-gpu-capacity#answer_705973

As per my knowledge, the fluctuation in the memory usage is due to the different computational and memory requirement across the layers of the U-Net network i.e., based on the layer properties there may be more number of computations involved for a layer when compared to another layer within the same network.

Refer to Deep Learning with Big Data on GPUs and in Parallel for other information.