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

refactor: docker images

Browse Source
This commit is contained in:
Félix Voituret
2019-11-21 12:25:25 +01:00
parent 5602d52041
commit 14b7eb2bab
19 changed files with 49 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docker/cpu/conda-2stems.dockerfile
View File

@@ -2,4 +2,5 @@ FROM researchdeezer/spleeter:conda
RUN mkdir -p /model/2stems \ RUN mkdir -p /model/2stems \
&& wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \ && wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \
&& tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ && tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ \
&& touch /model/5stems/.probe

3
docker/cpu/conda-4stems.dockerfile
View File

@@ -2,4 +2,5 @@ FROM researchdeezer/spleeter:conda
RUN mkdir -p /model/4stems \ RUN mkdir -p /model/4stems \
&& wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \ && wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \
&& tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ && tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ \
&& touch /model/5stems/.probe

3
docker/cpu/conda-5stems.dockerfile
View File

@@ -2,4 +2,5 @@ FROM researchdeezer/spleeter:conda
RUN mkdir -p /model/5stems \ RUN mkdir -p /model/5stems \
&& wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \ && wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \
&& tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ && tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ \
&& touch /model/5stems/.probe

2
docker/cpu/conda.dockerfile
View File

@@ -1,5 +1,7 @@
FROM continuumio/miniconda3:4.7.10 FROM continuumio/miniconda3:4.7.10
RUN conda install -y -c conda-forge musdb
# RUN conda install -y -c conda-forge museval
RUN conda install -y -c conda-forge spleeter RUN conda install -y -c conda-forge spleeter
RUN mkdir -p /model RUN mkdir -p /model
ENV MODEL_PATH /model ENV MODEL_PATH /model

24
docker/cpu/cpu.dockerfile
View File

@@ -1,24 +0,0 @@
FROM continuumio/miniconda3:4.7.10
# install tensorflow
RUN conda install -y tensorflow==1.14.0
# install ffmpeg for audio loading/writing
RUN conda install -y -c conda-forge ffmpeg
# install extra python libraries
RUN conda install -y -c anaconda pandas==0.25.1
RUN conda install -y -c conda-forge libsndfile
# install ipython
RUN conda install -y ipython
WORKDIR /workspace/
COPY ./ spleeter/
RUN mkdir /cache/
WORKDIR /workspace/spleeter
RUN pip install .
ENTRYPOINT ["python", "-m", "spleeter"]

5
docker/cpu/python-3.6-2stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.6 FROM researchdeezer/spleeter:3.6
RUN mkdir -p /model/2stems \ RUN mkdir -p /model/2stems \
&& wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \ && wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \
&& tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ && tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ \
&& touch /model/5stems/.probe

5
docker/cpu/python-3.6-4stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.6 FROM researchdeezer/spleeter:3.6
RUN mkdir -p /model/4stems \ RUN mkdir -p /model/4stems \
&& wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \ && wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \
&& tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ && tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ \
&& touch /model/5stems/.probe

5
docker/cpu/python-3.6-5stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.6 FROM researchdeezer/spleeter:3.6
RUN mkdir -p /model/5stems \ RUN mkdir -p /model/5stems \
&& wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \ && wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \
&& tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ && tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ \
&& touch /model/5stems/.probe

1
docker/cpu/python-3.6.dockerfile
View File

@@ -1,6 +1,7 @@
FROM python:3.6 FROM python:3.6
RUN apt-get update && apt-get install -y ffmpeg libsndfile RUN apt-get update && apt-get install -y ffmpeg libsndfile
RUN pip install musdb museval
RUN pip install spleeter RUN pip install spleeter
RUN mkdir -p /model RUN mkdir -p /model
ENV MODEL_PATH /model ENV MODEL_PATH /model

5
docker/cpu/python-3.7-2stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.7 FROM researchdeezer/spleeter:3.7
RUN mkdir -p /model/2stems \ RUN mkdir -p /model/2stems \
&& wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \ && wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \
&& tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ && tar -xvzf /tmp/2stems.tar.gz -C /model/2stems/ \
&& touch /model/5stems/.probe

5
docker/cpu/python-3.7-4stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.7 FROM researchdeezer/spleeter:3.7
RUN mkdir -p /model/4stems \ RUN mkdir -p /model/4stems \
&& wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \ && wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \
&& tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ && tar -xvzf /tmp/4stems.tar.gz -C /model/4stems/ \
&& touch /model/5stems/.probe

5
docker/cpu/python-3.7-5stems.dockerfile
View File

@@ -1,5 +1,6 @@
FROM spleeter:3.7 FROM researchdeezer/spleeter:3.7
RUN mkdir -p /model/5stems \ RUN mkdir -p /model/5stems \
&& wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \ && wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \
&& tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ && tar -xvzf /tmp/5stems.tar.gz -C /model/5stems/ \
&& touch /model/5stems/.probe

1
docker/cpu/python-3.7.dockerfile
View File

@@ -1,6 +1,7 @@
FROM python:3.7 FROM python:3.7
RUN apt-get update && apt-get install -y ffmpeg libsndfile RUN apt-get update && apt-get install -y ffmpeg libsndfile
RUN pip install musdb museval
RUN pip install spleeter RUN pip install spleeter
RUN mkdir -p /model RUN mkdir -p /model
ENV MODEL_PATH /model ENV MODEL_PATH /model

2
docker/gpu/conda-gpu-2-stems.dockerfile
View File

@@ -1,4 +1,4 @@
FROM deezer/spleeter:conda-gpu FROM researchdeezer/spleeter:conda-gpu
RUN mkdir -p /model/2stems \ RUN mkdir -p /model/2stems \
&& wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \ && wget -O /tmp/2stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz \

2
docker/gpu/conda-gpu-4-stems.dockerfile
View File

@@ -1,4 +1,4 @@
FROM deezer/spleeter:conda-gpu FROM researchdeezer/spleeter:conda-gpu
RUN mkdir -p /model/4stems \ RUN mkdir -p /model/4stems \
&& wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \ && wget -O /tmp/4stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/4stems.tar.gz \

2
docker/gpu/conda-gpu-5-stems.dockerfile
View File

@@ -1,4 +1,4 @@
FROM deezer/spleeter:conda-gpu FROM researchdeezer/spleeter:conda-gpu
RUN mkdir -p /model/5stems \ RUN mkdir -p /model/5stems \
&& wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \ && wget -O /tmp/5stems.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/5stems.tar.gz \

27
docker/gpu/conda-gpu.dockerfile
View File

@@ -1,12 +1,21 @@
FROM continuumio/miniconda3:4.7.10 FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
RUN conda install -y ipython \ RUN apt-get update --fix-missing && \
&& conda install -y tensorflow-gpu==1.14.0 \ apt-get install -y wget bzip2 ca-certificates curl git && \
&& conda install -y -c conda-forge ffmpeg \ apt-get clean && \
&& conda install -y -c conda-forge libsndfile \ rm -rf /var/lib/apt/lists/*
&& conda install -y -c anaconda pandas==0.25.1 \ RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-4.6.14-Linux-x86_64.sh -O ~/miniconda.sh && \
RUN mkdir -p /model /bin/bash ~/miniconda.sh -b -p /opt/conda && \
ENV MODEL_PATH /model rm ~/miniconda.sh && \
RUN pip install spleeter /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
RUN conda install -y tensorflow-gpu==1.14.0
RUN conda install -y -c conda-forge musdb
# RUN conda install -y -c conda-forge museval
# Note: switch to spleeter GPU once published.
RUN conda install -y -c conda-forge spleeter
ENTRYPOINT ["spleeter"] ENTRYPOINT ["spleeter"]

35
docker/gpu/gpu.dockerfile
View File

@@ -1,35 +0,0 @@
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
# set work directory
WORKDIR /workspace
# install anaconda
ENV PATH /opt/conda/bin:$PATH
COPY docker/install_miniconda.sh .
RUN bash ./install_miniconda.sh && rm install_miniconda.sh
RUN conda update -n base -c defaults conda
# install tensorflow for GPU
RUN conda install -y tensorflow-gpu==1.14.0
# install ffmpeg for audio loading/writing
RUN conda install -y -c conda-forge ffmpeg
# install extra libs
RUN conda install -y -c anaconda pandas==0.25.1
RUN conda install -y -c conda-forge libsndfile
# install ipython
RUN conda install -y ipython
RUN mkdir /cache/
# clone inside image github repository
COPY ./ spleeter/
WORKDIR /workspace/spleeter
RUN pip install .
ENTRYPOINT ["python", "-m", "spleeter"]

13
docker/gpu/install_miniconda.sh
View File

@@ -1,13 +0,0 @@
#!/bin/bash
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