pytorch / examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
https://pytorch.org/examples
BSD 3-Clause "New" or "Revised" License
22.21k stars 9.51k forks source link

examples/imagenet/main.py Multiple Gpus use for training #1136

Open Ansor-ZJJ opened 1 year ago

Ansor-ZJJ commented 1 year ago

By setting up multiple Gpus for use, the model and data are automatically loaded to these Gpus for training. What is the difference between this way and single-node multi-GPU distributed training?

alicera commented 1 year ago

https://pytorch.org/docs/stable/elastic/run.html Here is the information that you need