Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 465 Bytes

README.md

File metadata and controls

27 lines (20 loc) · 465 Bytes

DDP-example

Distributed training example.

Requirements

  • Python 3.8.6
pip install -r requirements.txt

Training

Single GPU training:

CUDA_VISIBLE_DEVICES=0 python train.py

Distributed training using two GPUs:

CUDA_VISIBLE_DEVICES=0,1 python train_ddp.py -g 2

Distributed training using two GPUs with Mixed Precision:

CUDA_VISIBLE_DEVICES=0,1 python train_ddp_mp.py -g 2