Skip to content

Updated runpod dockerfile to use miniforge and conda-forge package #22

Updated runpod dockerfile to use miniforge and conda-forge package

Updated runpod dockerfile to use miniforge and conda-forge package #22

name: publish-to-docker
on:
push:
branches:
- main
paths:
- pyproject.toml
- Dockerfile.cpu
- Dockerfile.gpu
- Dockerfile.runpod
workflow_dispatch:
jobs:
build-and-push-docker:
runs-on: ubuntu-latest
steps:
- name: Checkout
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v4
with:
python-version: "3.x"
- name: Install TOML
run: pip install toml
- name: Get version from pyproject.toml
run: |
VERSION=$(python -c "import toml; print(toml.load('pyproject.toml')['tool']['poetry']['version'])")
echo "VERSION=$VERSION" >> $GITHUB_ENV
- name: Set up QEMU
uses: docker/setup-qemu-action@v3
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Login to Docker Hub
uses: docker/login-action@v3
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
- name: Build and push Docker image for CPU
if: ${{ github.ref == 'refs/heads/main' }}
uses: docker/build-push-action@v5
with:
file: Dockerfile.cpu
context: .
platforms: linux/amd64,linux/arm64
push: true
tags: |
beveradb/audio-separator:cpu-${{ env.VERSION }}
beveradb/audio-separator:cpu
beveradb/audio-separator:latest
- name: Build and push Docker image for GPU
if: ${{ github.ref == 'refs/heads/main' }}
uses: docker/build-push-action@v5
with:
file: Dockerfile.gpu
context: .
platforms: linux/amd64
push: true
tags: |
beveradb/audio-separator:gpu-${{ env.VERSION }}
beveradb/audio-separator:gpu
# Deliberately commented out because Github CI can't build this runpod image due to disk space limits
# Instead, I build this (17GB) image locally and push it to Docker Hub manually.
# - name: Build and push Docker image for Runpod
# if: ${{ github.ref == 'refs/heads/main' }}
# uses: docker/build-push-action@v5
# with:
# file: Dockerfile.runpod
# context: .
# platforms: linux/amd64
# push: true
# tags: |
# beveradb/audio-separator:runpod-${{ env.VERSION }}
# beveradb/audio-separator:runpod