Using multi-gpu in minibatchqueue

2 次查看(过去 30 天)
Shaw
Shaw 2021-10-12
回答: Joss Knight 2021-10-24
I want to use multiple gpu's for my program. But, when I set the output environment of minibatchqueue to gpu, I guess , it is taking only one gpu.
There is no provision of using "multi-gpu" for outputenvironment. How can I use all of the GPU's using the minibatchqueue.

回答(2 个)

Ashutosh Singh Baghel
Hi Harsha,
I believe you wish to perform your calculations on multiple GPUs in parallel. Please refer to the following MATLAB Documentation page on 'Use Multiple GPUs in Parallel Pool' .

Joss Knight
Joss Knight 2021-10-24
https://uk.mathworks.com/help/deeplearning/ug/train-network-in-parallel-with-custom-training-loop.html will show you how to use minibatchqueue in conjunction with parallel syntax to partition your data and use data parallel training and inference.
minibatchqueue is a data iterator rather than a tool for managing the whole training loop like trainNetwork.

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

产品


版本

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by