Training on Multiple GPUs

https://d2l.ai/chapter_computational-performance/multiple-gpus.html