Add Dockerfile descriptions (#8031)
Browse files- utils/docker/Dockerfile +2 -0
- utils/docker/Dockerfile-arm64 +2 -1
- utils/docker/Dockerfile-cpu +2 -0
utils/docker/Dockerfile
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
|
|
|
|
|
|
| 2 |
|
| 3 |
# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 4 |
FROM nvcr.io/nvidia/pytorch:22.04-py3
|
|
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
| 2 |
+
# Builds ultralytics/yolov5:latest image on DockerHub https://hub.docker.com/r/ultralytics/yolov5
|
| 3 |
+
# Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference
|
| 4 |
|
| 5 |
# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 6 |
FROM nvcr.io/nvidia/pytorch:22.04-py3
|
utils/docker/Dockerfile-arm64
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
| 2 |
-
#
|
|
|
|
| 3 |
|
| 4 |
# Start FROM Ubuntu image https://hub.docker.com/_/ubuntu
|
| 5 |
FROM arm64v8/ubuntu:20.04
|
|
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
| 2 |
+
# Builds ultralytics/yolov5:latest-arm64 image on DockerHub https://hub.docker.com/r/ultralytics/yolov5
|
| 3 |
+
# Image is aarch64-compatible for Apple M1 and other ARM architectures i.e. Jetson Nano and Raspberry Pi
|
| 4 |
|
| 5 |
# Start FROM Ubuntu image https://hub.docker.com/_/ubuntu
|
| 6 |
FROM arm64v8/ubuntu:20.04
|
utils/docker/Dockerfile-cpu
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
|
|
|
|
|
|
| 2 |
|
| 3 |
# Start FROM Ubuntu image https://hub.docker.com/_/ubuntu
|
| 4 |
FROM ubuntu:20.04
|
|
|
|
| 1 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
| 2 |
+
# Builds ultralytics/yolov5:latest-cpu image on DockerHub https://hub.docker.com/r/ultralytics/yolov5
|
| 3 |
+
# Image is CPU-optimized for ONNX, OpenVINO and PyTorch YOLOv5 deployments
|
| 4 |
|
| 5 |
# Start FROM Ubuntu image https://hub.docker.com/_/ubuntu
|
| 6 |
FROM ubuntu:20.04
|