Skip to content

Commit

Permalink
remove pytorch build, only use cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
vemonet committed Oct 17, 2023
1 parent b2538b7 commit d0b613b
Showing 1 changed file with 0 additions and 68 deletions.
68 changes: 0 additions & 68 deletions .github/workflows/docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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/[email protected]
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}

0 comments on commit d0b613b

Please sign in to comment.