Does the example training script work for multi-gpu training?

materialsvirtuallab / matgl

Graph deep learning library for materials

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

233 stars 57 forks source link

Does the example training script work for multi-gpu training? #224

Closed matthewkuner closed 4 months ago

matthewkuner commented 5 months ago

https://github.com/materialsvirtuallab/matgl/blob/c99288e49f3bb83bf3381888581a877c88ce96d7/examples/Training%20a%20M3GNet%20Potential%20with%20PyTorch%20Lightning.ipynb#L1

Does the example training script for m3gnet potentials (linked) work for multi-gpu training? I'm specifically curious about fine-tuning it, though I'm guessing the answer is the same whether fine-tuning or training from scratch.

If multi-gpu training is not currently supported, are there plans to support it in the future?

shyuep commented 4 months ago

I would think it is no different. Multi-GPU is supported for sure.