From a05850286588600c632675bb128f628dc58b9070 Mon Sep 17 00:00:00 2001 From: nanaoto Date: Mon, 15 Sep 2025 17:47:08 +0800 Subject: [PATCH] Add Docker publish workflow configuration --- .github/workflows/docker-publish.yml | 44 ++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 .github/workflows/docker-publish.yml diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml new file mode 100644 index 0000000..f3e2572 --- /dev/null +++ b/.github/workflows/docker-publish.yml @@ -0,0 +1,44 @@ +name: Build and Publish Docker Image +on: + workflow_dispatch: + +jobs: + build-amd64: + runs-on: ubuntu-22.04 + strategy: + matrix: + include: + - cuda_version: 11.8 + torch_version: 2.4.1 + tag_prefix: pytorch2.4.1-cuda11.8 + - cuda_version: 12.8 + torch_version: 2.8.0 + tag_prefix: pytorch2.8.0-cuda12.8 + steps: + - name: Checkout code + uses: actions/checkout@v4 + + - name: Extract Docker Meta + id: meta + uses: docker/metadata-action@v5 + with: + images: nanaoto/index-tts + + - name: Set up Docker Buildx + uses: docker/setup-buildx-action@v3 + + - name: Build Docker Image + uses: docker/build-push-action@v5 + with: + context: . + file: ./Dockerfile + push: false + platforms: linux/amd64 + build-args: | + CUDA_VERSION=${{ matrix.cuda_version }} + TORCH_VERSION=${{ matrix.torch_version }} + tags: | + nanaoto/index-tts:${{ matrix.tag_prefix }}-${{ steps.meta.outputs.tags }}-amd64 + nanaoto/index-tts:latest-${{ matrix.tag_prefix }}-amd64 + +