Support training with multiple GPUs in same machine

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.

https://tryolabs.com

BSD 3-Clause "New" or "Revised" License

2.4k stars 399 forks source link

Support training with multiple GPUs in same machine #65

Open dekked opened 7 years ago

arun-kumark commented 6 years ago

Hi, I want to run the training on multiple GPU. But seems there is no control arguments to enable it. Please help, how to enable the multi-gpu training for Luminoth

vierja commented 6 years ago

Hi @arun-kumark, we currently only support using multiple GPUs using distributed training (which can be run on the same server).

We don't have any docs on how to use it yet but we support the same config as described in Google Cloud's doc.