Update Dockerfile (#4862) · 2809616345 - yolov5 - 拓恒技术版本管理系统

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

Unified View

							
								+ 3
								
								- 0
							
DockerfileView File
								
															# Update Ubuntu drivers
															
															# Update Ubuntu drivers
														
															# https://www.maketecheasier.com/install-nvidia-drivers-ubuntu/
															
															# 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