Fix docker image build (#19759)
* Use 2 jobs for the docker image build (latest torch + DS) * fix * Add comment Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
21
.github/workflows/build-docker-images.yml
vendored
21
.github/workflows/build-docker-images.yml
vendored
@@ -111,6 +111,27 @@ jobs:
|
|||||||
REF=main
|
REF=main
|
||||||
push: true
|
push: true
|
||||||
tags: huggingface/transformers-pytorch-deepspeed-latest-gpu${{ inputs.image_postfix }}
|
tags: huggingface/transformers-pytorch-deepspeed-latest-gpu${{ inputs.image_postfix }}
|
||||||
|
|
||||||
|
# Can't build 2 images in a single job `latest-torch-deepspeed-docker` (for `nvcr.io/nvidia`)
|
||||||
|
latest-torch-deepspeed-docker-for-push-ci-daily-build:
|
||||||
|
name: "Latest PyTorch + DeepSpeed (Push CI - Daily Build)"
|
||||||
|
# Can't run in parallel, otherwise get an error:
|
||||||
|
# `Error response from daemon: Get "https://registry-1.docker.io/v2/": received unexpected HTTP status: 503 Service Unavailable`
|
||||||
|
needs: latest-torch-deepspeed-docker
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
-
|
||||||
|
name: Set up Docker Buildx
|
||||||
|
uses: docker/setup-buildx-action@v1
|
||||||
|
-
|
||||||
|
name: Check out code
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
-
|
||||||
|
name: Login to DockerHub
|
||||||
|
uses: docker/login-action@v1
|
||||||
|
with:
|
||||||
|
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
||||||
|
password: ${{ secrets.DOCKERHUB_PASSWORD }}
|
||||||
# Push CI images still need to be re-built daily
|
# Push CI images still need to be re-built daily
|
||||||
-
|
-
|
||||||
name: Build and push (for Push CI) in a daily basis
|
name: Build and push (for Push CI) in a daily basis
|
||||||
|
|||||||
Reference in New Issue
Block a user