zjc
/
yolov5
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
Browse Source

Update Dockerfile (#4862)

modifyDataloader
Glenn Jocher GitHub 3 years ago
parent
40d1c80503
commit
2809616345
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +3
    -0
      Dockerfile

+ 3
- 0
Dockerfile View File

@@ -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

Write Preview
Loading…
Cancel
Save