| | 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 }} |
| |
|
