Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Update Dockerfile (#4862)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-20 13:10:41 +02:00
1 parent 40d1c80503
commit 2809616345
1 file changed
+3
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+3
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