yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ fix docker build

Browse Source
This commit is contained in:
Félix Voituret
2021-02-12 10:40:41 +01:00
parent 70150784dc
commit 69588a7be6
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
.github/workflows/docker.yml vendored
View File

@@ -1,13 +1,18 @@
name: docker name: docker
on: on:
- workflow_dispatch workflow_dispatch:
inputs:
version:
description: "Spleeter version to build image for"
required: true
default: "2.1.2"
jobs: jobs:
build-test-push: build-test-push:
runs-on: ubuntu-latest runs-on: ubuntu-latest
strategy: strategy:
matrix: matrix:
platform: [cpu, gpu] platform: [cpu, gpu]
distribution: [3.6, 3.7, 3.8, conda] distribution: [3.7, 3.8, conda]
model: [modelless, 2stems, 4stems, 5stems] model: [modelless, 2stems, 4stems, 5stems]
fail-fast: true fail-fast: true
steps: steps:
@@ -19,7 +24,6 @@ jobs:
echo "::set-env name=base::python:${{ matrix.distribution }}" echo "::set-env name=base::python:${{ matrix.distribution }}"
echo "::set-env name=tag::${{ matrix.distribution }}" echo "::set-env name=tag::${{ matrix.distribution }}"
echo "::set-env name=file::spleeter" echo "::set-env name=file::spleeter"
echo "::set-env name=package::spleeter"
- if: ${{ matrix.distribution == 'conda' }} - if: ${{ matrix.distribution == 'conda' }}
name: Build Conda base image name: Build Conda base image
run: | run: |
@@ -31,17 +35,16 @@ jobs:
docker build \ docker build \
--build-arg BASE=python:${{ matrix.distribution }} \ --build-arg BASE=python:${{ matrix.distribution }} \
-t cuda:${{ matrix.distribution }} \ -t cuda:${{ matrix.distribution }} \
-f docker/cuda-10-0.dockerfile . -f docker/cuda-10-1.dockerfile .
echo "::set-env name=base::cuda:${{ matrix.distribution }}" echo "::set-env name=base::cuda:${{ matrix.distribution }}"
echo "::set-env name=tag::${{ matrix.distribution }}-gpu" echo "::set-env name=tag::${{ matrix.distribution }}-gpu"
echo "::set-env name=package::spleeter-gpu"
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
# Note: image building. # Note: image building.
- name: Build deezer/spleeter:${{ env.tag }} image - name: Build deezer/spleeter:${{ env.tag }} image
run: | run: |
docker build \ docker build \
--build-arg BASE=${{ env.base }} \ --build-arg BASE=${{ env.base }} \
--build-arg SPLEETER_PACKAGE=${{ env.package }} \ --build-arg SPLEETER_VERSION=${{ github.event.inputs.version }} \
-t deezer/spleeter:${{ env.tag }} \ -t deezer/spleeter:${{ env.tag }} \
-f docker/${{ env.file }}.dockerfile . -f docker/${{ env.file }}.dockerfile .
echo "::set-env name=modelargs::" echo "::set-env name=modelargs::"
@@ -54,7 +57,7 @@ jobs:
-t deezer/spleeter:${{ env.tag }}-${{ matrix.model }} \ -t deezer/spleeter:${{ env.tag }}-${{ matrix.model }} \
-f docker/spleeter-model.dockerfile . -f docker/spleeter-model.dockerfile .
echo "::set-env name=tag::${{ env.tag }}-${{ matrix.model }}" echo "::set-env name=tag::${{ env.tag }}-${{ matrix.model }}"
echo "::set-env name=modelarg::-p spleeter:${{ matrix.model }}" echo "::set-env name=modelarg::-p spleeter:${{ matrix.model }}"
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
# Note: image testing. # Note: image testing.
- name: Test deezer/spleeter:${{ env.tag }} image - name: Test deezer/spleeter:${{ env.tag }} image