Train Deep Learning Model with multiple GPUs in Single Machine

Question

I'm using ArcGIS Pro and Deep Learning API version 3.2.

I was told that this version now supports training deep learning models using a single machine with multiple GPUs.  I have such a machine configured.  I selected in the Environment panel GPU as processor type and left the GPU ID as blank since there are multiple GPUs.

Is this how to instruct ArcGIS Pro Deep Learning to use multiple GPUs?  My model type is HED Edge Detector (Pixel Classification).  Does anyone know if HED is supported for multiple GPUs?  I have tried all possible model types and only HED came close to identifying field boundaries for my project.

Appreciate any response.

PriyankaTuteja · Answer

https://developers.arcgis.com/python/guide/utilize-multiple-gpus-to-train-model/

JadedEarth · Answer

Can you show me the link to instructions on how to run multi-gpu from command line?  There should have been instructions/guide to do this from the outset.  When you advertised that multi-gpu is supported, nowhere did it state that it only works for command line and there are no links to instructions on how to go about it.

PriyankaTuteja · Answer

Dear @JadedEarth

The multi-gpu support for deep learning models listed here is available.

Please note: The multi-gpu training is supported with command line only. If you are trying to use ArcGIS Pro tools, it would not work.