Which is the difference between 'multi-gpu' and 'parallel-gpu' in 'trainingOption' function of the DeepLearning Toolbox?

4 次查看（过去 30 天）
Enzo Losi
Enzo Losi 2024-5-22
回答： Joss Knight 2024-6-14
Hi everyone,
I have two NVIDIA RTX 3060 installed on my local computer and I want to train a neural network in parallel on both GPUs. I am worried about which is the best strategy between 'multi-gpu' and 'parallel-gpu'. Does anyone know how they work and which is the difference between 'multi-gpu' and 'parallel-gpu'?
Thank you.

请先登录，再回答此问题。

采纳的回答

Matt J
Matt J 2024-5-22
编辑：Matt J 2024-5-22
According to the doc, 'parallel-gpu' has the additional capability of being able to use remote GPUs. Since that doesn't apply to the hardware environment you describe, you can probably use either one.

更多回答（1 个）

Joss Knight
Joss Knight 2024-6-14
The purpose of 'multi-gpu' is effectively to try to ensure you are using a local pool with numGpus workers, rather than needing to understand anything about configuring a cluster. So either can work, but multi-gpu will give you helpful errors if you're doing something you didn't intend.

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

产品


版本

R2024a

Community Treasure Hunt

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

Start Hunting!

Translated by