CUDA Image #2
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: GPU Image | ||
| on: | ||
| workflow_dispatch: null | ||
| push: | ||
| paths: | ||
| - Dockerfile.cuda | ||
| - environment.yml | ||
| - install.r | ||
| jobs: | ||
| build: | ||
| runs-on: ubuntu-latest | ||
| permissions: write-all | ||
| steps: | ||
| # For biggish images, github actions runs out of disk space. | ||
| # So we cleanup some unwanted things in the disk image, and reclaim that space for our docker use | ||
| # https://github.com/actions/virtual-environments/issues/2606#issuecomment-772683150 | ||
| # and https://github.com/easimon/maximize-build-space/blob/b4d02c14493a9653fe7af06cc89ca5298071c66e/action.yml#L104 | ||
| # This gives us a total of about 52G of free space, which should be enough for now | ||
| - name: cleanup disk space | ||
| run: | | ||
| sudo rm -rf /usr/local/lib/android /usr/share/dotnet /opt/ghc | ||
| df -h | ||
| - uses: actions/checkout@v3 | ||
| - name: Login to GitHub Container Registry | ||
| if: github.repository == 'rocker-org/ml' | ||
| uses: docker/login-action@v1 | ||
| with: | ||
| registry: ghcr.io | ||
| username: ${{github.actor}} | ||
| password: ${{secrets.GITHUB_TOKEN}} | ||
| - name: Build the Docker image | ||
| if: github.repository == 'rocker-org/ml' | ||
| env: | ||
| - BASE: 'quay.io/jupyter/pytorch-notebook:cuda12-ubuntu-24.04' | ||
| run: docker build --no-cache . --tag ghcr.io/rocker/cuda:latest | ||
| - name: Publish | ||
| if: github.repository == 'rocker-org/ml' | ||
| run: docker push ghcr.io/rocker/cuda:latest | ||
