Commit
•
2809616
1
Parent(s):
40d1c80
Update Dockerfile (#4862)
Browse files- Dockerfile +3 -0
Dockerfile
CHANGED
|
@@ -53,3 +53,6 @@ ENV HOME=/usr/src/app
|
|
| 53 |
|
| 54 |
# Update Ubuntu drivers
|
| 55 |
# https://www.maketecheasier.com/install-nvidia-drivers-ubuntu/
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
# Update Ubuntu drivers
|
| 55 |
# https://www.maketecheasier.com/install-nvidia-drivers-ubuntu/
|
| 56 |
+
|
| 57 |
+
# DDP test
|
| 58 |
+
# python -m torch.distributed.run --nproc_per_node 2 --master_port 1 train.py --epochs 3
|