Demonstration of parallel atlas building

[jacobhinkle]

Is your feature request related to a problem? Please describe.
This is not really a problem but would make for a nice demo.

Describe the solution you'd like
It would be nice to have a jupyter notebook showing multi-gpu atlas building, as well as a python script showing multi-node multi-gpu atlas building. Affine would be fine.

Describe alternatives you've considered
NA

Additional context
NA

[rusheniii]

Hi Jacob,
Do you happen to have a notebook for parallel atlas building yet?

[jacobhinkle]

Nope this issue is still open. However, I've done this in https://github.com/jacobhinkle/diffeomorphic_autoencoders/blob/master/run_all.py
Note that that code is not using the lagomorph CLI, because I wrote that first and later put it into lagomorph. I think these days you should be able to do something like python -m torch.distributed.launch --nproc_per_node=4 lagomorph lddmm atlas .... If that does not work, we should fix it.