How to use multiple gpu on CGAN?
1 view (last 30 days)
I have used the code provided on mathworks website.
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?
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