How to use multiple gpu on CGAN?

1 vue (au cours des 30 derniers jours)
Shaw
Shaw le 16 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?

Réponses (2)

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

Catégories

En savoir plus sur GPU Computing dans Help Center et File Exchange

Produits


Version

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by