diff --git a/Dockerfile b/Dockerfile index 543e02e..95c098f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -53,3 +53,6 @@ ENV HOME=/usr/src/app # Update Ubuntu drivers # https://www.maketecheasier.com/install-nvidia-drivers-ubuntu/ + +# DDP test +# python -m torch.distributed.run --nproc_per_node 2 --master_port 1 train.py --epochs 3