From 28096163451a7bf3dc964228c557c3b8d010de2d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 20 Sep 2021 13:10:41 +0200 Subject: [PATCH] Update Dockerfile (#4862) --- Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Dockerfile b/Dockerfile index 543e02e34..95c098f9f 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