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

1 次查看（过去 30 天）
Huhufly
Huhufly 2021-5-19
评论： 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?

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

采纳的回答

Srivardhan Gadila
Srivardhan Gadila 2021-5-22
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.

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by