How to use multiple gpu on CGAN?

4 次查看（过去 30 天）
Shaw
Shaw 2021-10-16
回答： Nagwa megahed 2021-10-24
I have used the code provided on mathworks website.
" openExample('nnet/TrainConditionalGenerativeAdversarialNetworkCGANExample')"
It seems, though I have 2GPU's, the program is using only one GPU. How to use multi GPU on this program?
Where should I start the loop for sgmd?

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

回答（2 个）

Joss Knight
Joss Knight 2021-10-24
SPMD wraps the whole epoch loop. After you open it you partition the data and then start looping. See https://uk.mathworks.com/help/deeplearning/ug/train-network-in-parallel-with-custom-training-loop.html
Nagwa megahed
Nagwa megahed 2021-10-24
Firstly, you should install parallel computing toolbox
secondly ,try the code in this package Matlab-GAN

类别

在 Help Center 和 File Exchange 中查找有关 Startup and Shutdown 的更多信息

标签

产品


版本

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by