How to use multiple gpu on CGAN?

1 回表示 (過去 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 で GPU Computing についてさらに検索

タグ

製品


リリース

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by