Update Dockerfile (#4862)

2021-09-20 13:10:41 +02:00 · 2021-09-20 13:10:41 +02:00 · 2809616345
parent 40d1c80503
commit 2809616345
1 changed files with 3 additions and 0 deletions
--- a/3
+++ b/3
@ -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