diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index 62ff0bd..216de94 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -69,71 +69,3 @@ jobs: TAGS: ${{ steps.meta.outputs.tags }} DIGEST: ${{ steps.build-and-push.outputs.digest }} run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST} - - - - publish-docker-pytorch: - name: 🐳 Publish PyTorch GPU docker image - runs-on: ubuntu-latest - permissions: - contents: read - packages: write - id-token: write # This is used to complete the identity challenge with sigstore/fulcio - env: - REGISTRY: ghcr.io - IMAGE_NAME: ${{ github.repository }} - - steps: - - name: Checkout repository - uses: actions/checkout@v3 - - - name: Install cosign - if: github.event_name != 'pull_request' - uses: sigstore/cosign-installer@v3.1.1 - with: - cosign-release: 'v2.1.1' - - - name: Setup Docker buildx - uses: docker/setup-buildx-action@v2 - - - name: Log into registry ${{ env.REGISTRY }} - if: github.event_name != 'pull_request' - uses: docker/login-action@v2 - with: - registry: ${{ env.REGISTRY }} - username: ${{ github.actor }} - password: ${{ secrets.GITHUB_TOKEN }} - - - name: Extract Docker metadata - id: meta - uses: docker/metadata-action@v4 - with: - images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }} - tags: | - type=ref,event=branch,suffix=-gpu - type=semver,pattern={{version}},suffix=-gpu - - - name: Build and push Docker image - id: build-and-push - uses: docker/build-push-action@v4 - with: - context: . - file: Dockerfile.gpu - build-args: | - BASE_IMAGE=nvcr.io/nvidia/cuda:12.2.0-devel-ubuntu22.04 - GPU_ENABLED=true - # BASE_IMAGE=nvcr.io/nvidia/pytorch:23.06-py3 - # BASE_IMAGE=nvcr.io/nvidia/cuda:12.2.0-devel-ubuntu22.04 - push: ${{ github.event_name != 'pull_request' }} - tags: ${{ steps.meta.outputs.tags }} - labels: ${{ steps.meta.outputs.labels }} - cache-from: type=gha - cache-to: type=gha,mode=max - - # Sign the resulting Docker image digest except on PRs. - - name: Sign the published Docker image - if: ${{ github.event_name != 'pull_request' }} - env: - TAGS: ${{ steps.meta.outputs.tags }} - DIGEST: ${{ steps.build-and-push.outputs.digest }} - run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST}