How to use multiple gpu on CGAN?

7 views (last 30 days)
Shaw
Shaw on 16 Oct 2021
Answered: Nagwa megahed on 24 Oct 2021
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?

Answers (2)

Joss Knight
Joss Knight on 24 Oct 2021
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 on 24 Oct 2021
Firstly, you should install parallel computing toolbox
secondly ,try the code in this package Matlab-GAN

Products


Release

R2021a

Community Treasure Hunt

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

Start Hunting!