Multi-GPU for training

[fabian247]

Hi.
I'm trying to train for a specific 3D model using this repository.

The first try uses the lighter demo with a different 3D model.

It seems to work, but training is a bit slow.
I have 4 GPUs in my render machine, but only one of them is used for training. Can I enable all 4 GPUs in the training script or a setting somehow?

Best regards,

Fabian

[xavierjs]

Hi,

Since WebAR.rocks.train uses WebGL, it is more a generalistic WebGL/browser/GPU issue.
With 4 GPUs you can either train 4 neural nets in parallel with different parameters to pick the best one, or maybe you can bind them together with SLI or something like that to get 1 big virtual GPU, but it is out of scope of this code repo. Not sure though.

[fabian247]

Thanks for the quick reply. I'll see if I have SLI or NVLink available. Might not be the case though.