diff --git a/.circleci/config.yml b/.circleci/config.yml index 5d4acc1..16b71ea 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -91,16 +91,18 @@ jobs: # ======================================================================================= # Docker build. # ======================================================================================= + docker-conda-cpu: docker: - image: docker:17.05.0-ce-git steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:conda -f docker/conda.dockerfile . - - run: docker build --build-arg BASETAG=conda --build-arg MODEL=2stems -t researchdeezer/spleeter:conda-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=conda --build-arg MODEL=4stems -t researchdeezer/spleeter:conda-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=conda --build-arg MODEL=5stems -t researchdeezer/spleeter:conda-5stems -f docker/embedded-model.dockerfile . + - run: docker build -t conda -f docker/conda.dockerfile . + - run: docker build --build-arg BASE=conda -t researchdeezer/spleeter:conda -f docker/spleeter-conda.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda --build-arg MODEL=2stems -t researchdeezer/spleeter:conda-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda --build-arg MODEL=4stems -t researchdeezer/spleeter:conda-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda --build-arg MODEL=5stems -t researchdeezer/spleeter:conda-5stems -f docker/spleeter-model.dockerfile . - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:conda separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:conda-2stems separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:conda-4stems separate -i /runtime/audio_example.mp3 -p spleeter:4stems -o /tmp @@ -116,10 +118,13 @@ jobs: steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:conda-gpu -f docker/conda-gpu.dockerfile . - - run: docker build --build-arg BASETAG=conda-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:conda-gpu-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=conda-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:conda-gpu-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=conda-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:conda-gpu-5stems -f docker/embedded-model.dockerfile . + - run: docker build -t conda -f docker/conda.dockerfile . + - run: docker build --build-arg BASE=conda -t cuda:conda -f docker/cuda-10-0.dockerfile . + - run: docker build --build-arg BASE=cuda:conda -t conda-gpu -f docker/conda.dockerfile . + - run: docker build --build-arg BASE=conda-gpu -t researchdeezer/spleeter:conda-gpu -f docker/spleeter-conda.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:conda-gpu-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:conda-gpu-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:conda-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:conda-gpu-5stems -f docker/spleeter-model.dockerfile . - run: docker login -u $DOCKERHUB_USERNAME -p $DOCKERHUB_PASSWORD - run: docker push researchdeezer/spleeter:conda-gpu - run: docker push researchdeezer/spleeter:conda-gpu-2stems @@ -133,10 +138,10 @@ jobs: steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:3.6 -f docker/python-3.6.dockerfile . - - run: docker build --build-arg BASETAG=3.6 --build-arg MODEL=2stems -t researchdeezer/spleeter:3.6-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.6 --build-arg MODEL=4stems -t researchdeezer/spleeter:3.6-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.6 --build-arg MODEL=5stems -t researchdeezer/spleeter:3.6-5stems -f docker/embedded-model.dockerfile . + - run: docker build --build-arg BASE=python:3.6 -t researchdeezer/spleeter:3.6 -f docker/spleeter.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6 --build-arg MODEL=2stems -t researchdeezer/spleeter:3.6-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6 --build-arg MODEL=4stems -t researchdeezer/spleeter:3.6-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6 --build-arg MODEL=5stems -t researchdeezer/spleeter:3.6-5stems -f docker/spleeter-model.dockerfile . - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.6 separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.6-2stems separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.6-4stems separate -i /runtime/audio_example.mp3 -p spleeter:4stems -o /tmp @@ -152,10 +157,11 @@ jobs: steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:3.6-gpu -f docker/python-3.6-gpu.dockerfile . - - run: docker build --build-arg BASETAG=3.6-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:3.6-gpu-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.6-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:3.6-gpu-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.6-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:3.6-gpu-5stems -f docker/embedded-model.dockerfile . + - run: docker build --build-arg BASE=python:3.6 -t cuda:3.6 -f docker/cuda-10-0.dockerfile . + - run: docker build --build-arg BASE=cuda:3.6 -t researchdeezer/spleeter:3.6-gpu -f docker/spleeter.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:3.6-gpu-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:3.6-gpu-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.6-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:3.6-gpu-5stems -f docker/spleeter-model.dockerfile . - run: docker login -u $DOCKERHUB_USERNAME -p $DOCKERHUB_PASSWORD - run: docker push researchdeezer/spleeter:3.6-gpu - run: docker push researchdeezer/spleeter:3.6-gpu-2stems @@ -167,10 +173,10 @@ jobs: steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:3.7 -f docker/python-3.7.dockerfile . - - run: docker build --build-arg BASETAG=3.7 --build-arg MODEL=2stems -t researchdeezer/spleeter:3.7-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.7 --build-arg MODEL=4stems -t researchdeezer/spleeter:3.7-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.7 --build-arg MODEL=5stems -t researchdeezer/spleeter:3.7-5stems -f docker/embedded-model.dockerfile . + - run: docker build --build-arg BASE=python:3.7 -t researchdeezer/spleeter:3.7 -f docker/spleeter.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7 --build-arg MODEL=2stems -t researchdeezer/spleeter:3.7-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7 --build-arg MODEL=4stems -t researchdeezer/spleeter:3.7-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7 --build-arg MODEL=5stems -t researchdeezer/spleeter:3.7-5stems -f docker/spleeter-model.dockerfile . - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.7 separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.7-2stems separate -i /runtime/audio_example.mp3 -o /tmp - run: docker run -v $(pwd):/runtime researchdeezer/spleeter:3.7-4stems separate -i /runtime/audio_example.mp3 -p spleeter:4stems -o /tmp @@ -188,15 +194,17 @@ jobs: steps: - checkout - setup_remote_docker - - run: docker build -t researchdeezer/spleeter:3.7-gpu -f docker/python-3.7-gpu.dockerfile . - - run: docker build --build-arg BASETAG=3.7-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:3.7-gpu-2stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.7-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:3.7-gpu-4stems -f docker/embedded-model.dockerfile . - - run: docker build --build-arg BASETAG=3.7-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:3.7-gpu-5stems -f docker/embedded-model.dockerfile . + - run: docker build -t cuda:3.7 -f docker/cuda-10-0.dockerfile . + - run: docker build --build-arg BASE=cuda:3.7 -t researchdeezer/spleeter:3.7-gpu -f docker/spleeter.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7-gpu --build-arg MODEL=2stems -t researchdeezer/spleeter:3.7-gpu-2stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7-gpu --build-arg MODEL=4stems -t researchdeezer/spleeter:3.7-gpu-4stems -f docker/spleeter-model.dockerfile . + - run: docker build --build-arg BASE=researchdeezer/spleeter:3.7-gpu --build-arg MODEL=5stems -t researchdeezer/spleeter:3.7-gpu-5stems -f docker/spleeter-model.dockerfile . - run: docker login -u $DOCKERHUB_USERNAME -p $DOCKERHUB_PASSWORD - run: docker push researchdeezer/spleeter:3.7-gpu - run: docker push researchdeezer/spleeter:3.7-gpu-2stems - run: docker push researchdeezer/spleeter:3.7-gpu-4stems - run: docker push researchdeezer/spleeter:3.7-gpu-5stems + workflows: version: 2 spleeter-release-pipeline: diff --git a/docker/conda-gpu.dockerfile b/docker/conda-gpu.dockerfile deleted file mode 100644 index 2fb92d7..0000000 --- a/docker/conda-gpu.dockerfile +++ /dev/null @@ -1,23 +0,0 @@ -FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 - -RUN apt-get update --fix-missing \ - && apt-get install -y wget bzip2 ca-certificates curl git \ - && apt-get clean \ - && rm -rf /var/lib/apt/lists/* \ - && wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-4.6.14-Linux-x86_64.sh -O ~/miniconda.sh \ - && /bin/bash ~/miniconda.sh -b -p /opt/conda \ - && rm ~/miniconda.sh \ - && /opt/conda/bin/conda clean -tipsy \ - && ln -s /opt/conda/etc/profile.d/conda.sh /etc/profile.d/conda.sh \ - && echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc \ - && echo "conda activate base" >> ~/.bashrc \ - && ln -s /opt/conda/bin/conda /usr/bin/conda - -COPY audio_example.mp3 . - -RUN conda install -y -c conda-forge musdb -# RUN conda install -y -c conda-forge museval -RUN conda install -y -c conda-forge spleeter-gpu=1.4.9 - - -ENTRYPOINT ["spleeter"] \ No newline at end of file diff --git a/docker/conda.dockerfile b/docker/conda.dockerfile index 1ecf193..349cbae 100644 --- a/docker/conda.dockerfile +++ b/docker/conda.dockerfile @@ -1,11 +1,15 @@ -FROM continuumio/miniconda3:4.7.10 +ARG BASE=python:3.7 +FROM ${BASE} -RUN mkdir -p /model -ENV MODEL_PATH /model -COPY audio_example.mp3 . - -RUN conda install -y -c conda-forge musdb -# RUN conda install -y -c conda-forge museval -RUN conda install -y -c conda-forge spleeter=1.4.9 - -ENTRYPOINT ["spleeter"] \ No newline at end of file +RUN apt-get update --fix-missing \ + && apt-get install -y wget bzip2 ca-certificates curl git \ + && apt-get clean \ + && rm -rf /var/lib/apt/lists/* \ + && wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-4.6.14-Linux-x86_64.sh -O ~/miniconda.sh \ + && /bin/bash ~/miniconda.sh -b -p /opt/conda \ + && rm ~/miniconda.sh \ + && /opt/conda/bin/conda clean -tipsy \ + && ln -s /opt/conda/etc/profile.d/conda.sh /etc/profile.d/conda.sh \ + && echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc \ + && echo "conda activate base" >> ~/.bashrc \ + && ln -s /opt/conda/bin/conda /usr/bin/conda diff --git a/docker/cuda-10-0.dockerfile b/docker/cuda-10-0.dockerfile new file mode 100644 index 0000000..8c7af31 --- /dev/null +++ b/docker/cuda-10-0.dockerfile @@ -0,0 +1,43 @@ +ARG BASE=python:3.7 +FROM ${BASE} + +ENV CUDA_VERSION 10.0.130 +ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1 +ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} +ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 +# Note: differ from original image, picked from old dzr_audio spec. +ENV NVIDIA_VISIBLE_DEVICES=3 +ENV NVIDIA_DRIVER_CAPABILITIES compute,utility +ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411" +ENV NCCL_VERSION 2.4.2 +ENV CUDNN_VERSION 7.6.0.64 + +LABEL com.nvidia.cuda.version="${CUDA_VERSION}" +LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}" +LABEL com.nvidia.volumes.needed="nvidia_driver" + +RUN apt-get update \ + && apt-get install -y --no-install-recommends \ + gnupg2 \ + curl \ + ca-certificates \ + && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \ + && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list \ + && echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list \ + && apt-get purge --autoremove -y curl \ + && apt-get update \ + && apt-get install -y --no-install-recommends \ + cuda-cudart-$CUDA_PKG_VERSION \ + cuda-compat-10-0 \ + && ln -s cuda-10.0 /usr/local/cuda \ + && echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf \ + && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf \ + && apt-get install -y --no-install-recommends \ + cuda-toolkit-10-0 \ + cuda-libraries-$CUDA_PKG_VERSION \ + cuda-nvtx-$CUDA_PKG_VERSION \ + libnccl2=$NCCL_VERSION-1+cuda10.0 \ + libcudnn7=$CUDNN_VERSION-1+cuda10.0 \ + && apt-mark hold libnccl2 \ + && apt-mark hold libcudnn7 \ + && rm -rf /var/lib/apt/lists/* diff --git a/docker/python-3.7-gpu.dockerfile b/docker/cuda-9.2.dockerfile similarity index 98% rename from docker/python-3.7-gpu.dockerfile rename to docker/cuda-9.2.dockerfile index e4e4de6..06eecb4 100644 --- a/docker/python-3.7-gpu.dockerfile +++ b/docker/cuda-9.2.dockerfile @@ -1,4 +1,5 @@ -FROM python:3.7 +ARG BASE=python:3.7 +FROM ${BASE} # FROM 9.2-base-ubuntu18.04 # https://gitlab.com/nvidia/container-images/cuda/blob/ubuntu18.04/9.2/base/Dockerfile diff --git a/docker/python-3.6-gpu.dockerfile b/docker/python-3.6-gpu.dockerfile deleted file mode 100644 index 4fcd596..0000000 --- a/docker/python-3.6-gpu.dockerfile +++ /dev/null @@ -1,57 +0,0 @@ -FROM python:3.6 - -# FROM 9.2-base-ubuntu18.04 -# https://gitlab.com/nvidia/container-images/cuda/blob/ubuntu18.04/9.2/base/Dockerfile -RUN apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates \ - && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64/7fa2af80.pub | apt-key add - \ - && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64 /" > /etc/apt/sources.list.d/cuda.list \ - && echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list \ - && apt-get purge --autoremove -y curl \ - && rm -rf /var/lib/apt/lists/* -ENV CUDA_VERSION 9.2.148 -ENV CUDA_PKG_VERSION 9-2=$CUDA_VERSION-1 -RUN apt-get update \ - && apt-get install -y --no-install-recommends \ - cuda-cudart-$CUDA_PKG_VERSION \ - && ln -s cuda-9.2 /usr/local/cuda \ - && rm -rf /var/lib/apt/lists/* -LABEL com.nvidia.volumes.needed="nvidia_driver" -LABEL com.nvidia.cuda.version="${CUDA_VERSION}" -RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf \ - && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf -ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} -ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 -ENV NVIDIA_VISIBLE_DEVICES all -ENV NVIDIA_DRIVER_CAPABILITIES compute,utility -ENV NVIDIA_REQUIRE_CUDA "cuda>=9.2" - -# FROM 9.2-runtime-ubuntu18.04 -# https://gitlab.com/nvidia/container-images/cuda/blob/ubuntu18.04/9.2/runtime/Dockerfile -ENV NCCL_VERSION 2.3.7 -RUN apt-get update \ - && apt-get install -y --no-install-recommends \ - cuda-libraries-$CUDA_PKG_VERSION \ - cuda-nvtx-$CUDA_PKG_VERSION \ - libnccl2=$NCCL_VERSION-1+cuda9.2 \ - && apt-mark hold libnccl2 \ - && rm -rf /var/lib/apt/lists/* - -# FROM 9.2-runtime-cudnn7-ubuntu18.04 -# https://gitlab.com/nvidia/container-images/cuda/blob/ubuntu18.04/9.2/runtime/cudnn7/Dockerfile -ENV CUDNN_VERSION 7.5.0.56 -LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}" -RUN apt-get update \ - && apt-get install -y --no-install-recommends libcudnn7=$CUDNN_VERSION-1+cuda9.2 \ - && apt-mark hold libcudnn7 \ - && rm -rf /var/lib/apt/lists/* - -RUN mkdir -p /model -ENV MODEL_PATH /model -COPY audio_example.mp3 . - -# Spleeter installation. -RUN apt-get update && apt-get install -y ffmpeg libsndfile1 -RUN pip install musdb museval -RUN pip install spleeter-gpu==1.4.9 - -ENTRYPOINT ["spleeter"] \ No newline at end of file diff --git a/docker/python-3.6.dockerfile b/docker/python-3.6.dockerfile deleted file mode 100644 index 0cad379..0000000 --- a/docker/python-3.6.dockerfile +++ /dev/null @@ -1,11 +0,0 @@ -FROM python:3.6 - -RUN mkdir -p /model -ENV MODEL_PATH /model -COPY audio_example.mp3 . - -RUN apt-get update && apt-get install -y ffmpeg libsndfile1 -RUN pip install musdb museval -RUN pip install spleeter==1.4.9 - -ENTRYPOINT ["spleeter"] \ No newline at end of file diff --git a/docker/spleeter-conda.dockerfile b/docker/spleeter-conda.dockerfile new file mode 100644 index 0000000..da8b90a --- /dev/null +++ b/docker/spleeter-conda.dockerfile @@ -0,0 +1,15 @@ +ARG BASE=conda +ARG SPLEETER_PACKAGE=spleeter +ARG SPLEETER_VERSION=1.4.9 + +FROM ${BASE} + +ENV MODEL_PATH /model + +RUN mkdir -p /model +COPY audio_example.mp3 . + +RUN conda install -y -c conda-forge musdb +RUN conda install -y -c conda-forge ${SPLEETER_PACKAGE}==${SPLEETER_VERSION} + +ENTRYPOINT ["spleeter"] \ No newline at end of file diff --git a/docker/embedded-model.dockerfile b/docker/spleeter-model.dockerfile similarity index 78% rename from docker/embedded-model.dockerfile rename to docker/spleeter-model.dockerfile index 4795bef..7bdffc4 100644 --- a/docker/embedded-model.dockerfile +++ b/docker/spleeter-model.dockerfile @@ -1,8 +1,7 @@ -ARG BASETAG +ARG BASE=researchdeezer/spleeter +ARG MODEL=2-stems -FROM researchdeezer/spleeter:${BASETAG} - -ARG MODEL +FROM ${BASE} RUN mkdir -p /model/$MODEL \ && wget -O /tmp/$MODEL.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/$MODEL.tar.gz \ diff --git a/docker/python-3.7.dockerfile b/docker/spleeter.dockerfile similarity index 55% rename from docker/python-3.7.dockerfile rename to docker/spleeter.dockerfile index efa9c1f..a28c2ee 100644 --- a/docker/python-3.7.dockerfile +++ b/docker/spleeter.dockerfile @@ -1,11 +1,16 @@ -FROM python:3.7 +ARG BASE=python:3.6 +ARG SPLEETER_PACKAGE=spleeter +ARG SPLEETER_VERSION=1.4.9 + +FROM ${BASE} + +ENV MODEL_PATH /model RUN mkdir -p /model -ENV MODEL_PATH /model COPY audio_example.mp3 . RUN apt-get update && apt-get install -y ffmpeg libsndfile1 RUN pip install musdb museval -RUN pip install spleeter==1.4.9 +RUN pip install ${SPLEETER_PACKAGE}==${SPLEETER_VERSION} ENTRYPOINT ["spleeter"] \ No newline at end of file