mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
35 lines
770 B
Docker
35 lines
770 B
Docker
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"] |