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
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' .

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 Center および File ExchangeImage Data Workflows についてさらに検索

製品


リリース

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by