Skip to content

Docker Quickstart

Glenn Jocher edited this page Dec 14, 2021 · 54 revisions

To get started with YOLOv5 ๐Ÿš€ in a Docker image follow the instructions below. Other quickstart options for YOLOv5 include our Colab Notebook Open In Colab Open In Kaggle and a GCP Deep Learning VM.

1. Install Docker and Nvidia-Docker

Docker images come with all dependencies preinstalled, however Docker itself requires installation, and relies of nvidia driver installations in order to interact properly with local GPU resources. The requirements are:

2. Pull Image

The Ultralytics YOLOv5 DockerHub is https://hub.docker.com/r/ultralytics/yolov5 Docker Pulls. Docker Autobuild is used to automatically build images from the latest repository commits, so the ultralytics/yolov5:latest image hosted on the DockerHub will always be in sync with the most recent repository commit. To pull this image:

sudo docker pull ultralytics/yolov5:latest

3. Run Container

Run an interactive instance of this image (called a "container") using -it:

sudo docker run --ipc=host -it ultralytics/yolov5:latest

Run a container with local file access (like COCO training data in /datasets) using -v:

sudo docker run --ipc=host -it -v "$(pwd)"/datasets:/usr/src/datasets ultralytics/yolov5:latest

Run a container with GPU access using --gpus all:

sudo docker run --ipc=host -it --gpus all ultralytics/yolov5:latest

4. Run YOLOv5 ๐Ÿš€

Start training, testing, detecting and exporting YOLOv5 models within the running Docker container!

$ python train.py  # train a model
$ python val.py --weights yolov5s.pt  # validate a model for Precision, Recall and mAP
$ python detect.py --weights yolov5s.pt --source path/to/images  # run inference on images and videos
$ python export.py --weights yolov5s.pt --include onnx coreml tflite  # export models to other formats


ultralytics/yolov5:latest-cpu Image

FROM ubuntu:latest

# Install linux packages
RUN apt update && apt install -y python3-pip zip htop screen libgl1-mesa-glx
RUN alias python=python3

# Install python dependencies
COPY requirements.txt .
RUN python3 -m pip install --no-cache --upgrade pip
RUN pip uninstall -y nvidia-tensorboard nvidia-tensorboard-plugin-dlprof
RUN pip install --no-cache -r requirements.txt coremltools onnx onnxruntime tensorflow-cpu gsutil notebook
RUN pip install --no-cache -U numpy Pillow
RUN pip install --no-cache torch==1.10.0+cpu torchvision==0.11.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
# RUN pip install --no-cache torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

# Create working directory
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app

# Copy contents
COPY . /usr/src/app

# Downloads to user config dir
ADD https://ultralytics.com/assets/Arial.ttf /root/.config/Ultralytics/
Clone this wiki locally