|
|
|
|
|
|
|
name: Publish Docker Images |
|
|
|
on: |
|
push: |
|
branches: [ master ] |
|
|
|
jobs: |
|
docker: |
|
if: github.repository == 'ultralytics/yolov5' |
|
name: Push Docker image to Docker Hub |
|
runs-on: ubuntu-latest |
|
steps: |
|
- name: Checkout repo |
|
uses: actions/checkout@v3 |
|
|
|
- name: Set up QEMU |
|
uses: docker/setup-qemu-action@v2 |
|
|
|
- name: Set up Docker Buildx |
|
uses: docker/setup-buildx-action@v2 |
|
|
|
- name: Login to Docker Hub |
|
uses: docker/login-action@v2 |
|
with: |
|
username: ${{ secrets.DOCKERHUB_USERNAME }} |
|
password: ${{ secrets.DOCKERHUB_TOKEN }} |
|
|
|
- name: Build and push arm64 image |
|
uses: docker/build-push-action@v3 |
|
with: |
|
context: . |
|
platforms: linux/arm64 |
|
file: utils/docker/Dockerfile-arm64 |
|
push: true |
|
tags: ultralytics/yolov5:latest-arm64 |
|
|
|
- name: Build and push CPU image |
|
uses: docker/build-push-action@v3 |
|
with: |
|
context: . |
|
file: utils/docker/Dockerfile-cpu |
|
push: true |
|
tags: ultralytics/yolov5:latest-cpu |
|
|
|
- name: Build and push GPU image |
|
uses: docker/build-push-action@v3 |
|
with: |
|
context: . |
|
file: utils/docker/Dockerfile |
|
push: true |
|
tags: ultralytics/yolov5:latest |
|
|
