Using multi-gpu in minibatchqueue
5 次查看(过去 30 天)
显示 更早的评论
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.
0 个评论
回答(2 个)
Ashutosh Singh Baghel
2021-10-20
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' .
0 个评论
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.
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!