| name: Docker |
| on: |
| push: |
| branches: |
| - main |
| workflow_dispatch: {} |
| jobs: |
| docker-build: |
| runs-on: ubuntu-latest |
| if: github.repository_owner == 'mosaicml' |
| strategy: |
| matrix: |
| include: |
| - name: '1.13.1_cu117' |
| base_image: mosaicml/pytorch:1.13.1_cu117-python3.10-ubuntu20.04 |
| - name: '2.0.1_cu118' |
| base_image: mosaicml/pytorch:2.0.1_cu118-python3.10-ubuntu20.04 |
|
|
| steps: |
| - name: Maximize Build Space on Worker |
| uses: easimon/maximize-build-space@v4 |
| with: |
| overprovision-lvm: true |
| remove-dotnet: true |
| remove-android: true |
| remove-haskell: true |
|
|
| - name: Checkout |
| uses: actions/checkout@v3 |
|
|
| - name: Setup QEMU |
| uses: docker/setup-qemu-action@v2 |
|
|
| - name: Setup Docker Buildx |
| uses: docker/setup-buildx-action@v2 |
|
|
| - name: Login to DockerHub |
| uses: docker/login-action@v2 |
| with: |
| username: ${{ secrets.DOCKER_HUB_USERNAME }} |
| password: ${{ secrets.DOCKER_HUB_PASSWORD }} |
|
|
| - name: Calculate Docker Image Variables |
| run: | |
| set -euxo pipefail |
| |
| |
| |
| |
| GIT_SHA=$(echo ${{ github.sha }} | cut -c1-7) |
| echo "IMAGE_TAG=${GIT_SHA}" >> ${GITHUB_ENV} |
|
|
| - name: Build and Push the Docker Image |
| uses: docker/build-push-action@v3 |
| with: |
| context: . |
| tags: mosaicml/llm-foundry:${{ matrix.name }}-latest, |
| mosaicml/llm-foundry:${{ matrix.name }}-${{ env.IMAGE_TAG }} |
| push: true |
| cache-from: type=registry,ref=mosaicml/llm-foundry:${{ matrix.name }}-buildcache |
| cache-to: type=registry,ref=mosaicml/llm-foundry:${{ matrix.name }}-buildcache,mode=max |
| build-args: BASE_IMAGE=${{ matrix.base_image }} |
|
|
