How to use multiple gpu on CGAN?

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?

Answer 1

[Joss Knight]

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

Answer 2

[Nagwa megahed]

Firstly, you should install parallel computing toolbox

secondly ,try the code in this package Matlab-GAN