You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.3 KiB
81 lines
2.3 KiB
# Ultralytics YOLO 🚀, AGPL-3.0 license |
|
# Builds ultralytics/ultralytics:latest images on DockerHub https://hub.docker.com/r/ultralytics |
|
|
|
name: Publish Docker Images |
|
|
|
on: |
|
push: |
|
branches: [main] |
|
workflow_dispatch: |
|
inputs: |
|
image: |
|
type: choice |
|
description: Select Docker Image |
|
options: |
|
- Arm64 |
|
- Jetson |
|
- CPU |
|
- GPU |
|
|
|
jobs: |
|
docker: |
|
if: github.repository == 'ultralytics/ultralytics' |
|
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 |
|
if: github.event_name == 'push' || github.event.inputs.image == 'Arm64' |
|
uses: docker/build-push-action@v4 |
|
continue-on-error: true |
|
with: |
|
context: . |
|
platforms: linux/arm64 |
|
file: docker/Dockerfile-arm64 |
|
push: true |
|
tags: ultralytics/ultralytics:latest-arm64 |
|
|
|
- name: Build and push Jetson image |
|
if: github.event_name == 'push' || github.event.inputs.image == 'Jetson' |
|
uses: docker/build-push-action@v4 |
|
continue-on-error: true |
|
with: |
|
context: . |
|
platforms: linux/arm64 |
|
file: docker/Dockerfile-jetson |
|
push: true |
|
tags: ultralytics/ultralytics:latest-jetson |
|
|
|
- name: Build and push CPU image |
|
if: github.event_name == 'push' || github.event.inputs.image == 'CPU' |
|
uses: docker/build-push-action@v4 |
|
continue-on-error: true |
|
with: |
|
context: . |
|
file: docker/Dockerfile-cpu |
|
push: true |
|
tags: ultralytics/ultralytics:latest-cpu |
|
|
|
- name: Build and push GPU image |
|
if: github.event_name == 'push' || github.event.inputs.image == 'GPU' |
|
uses: docker/build-push-action@v4 |
|
continue-on-error: true |
|
with: |
|
context: . |
|
file: docker/Dockerfile |
|
push: true |
|
tags: ultralytics/ultralytics:latest
|
|
|