mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
Initial commit from private spleeter
This commit is contained in:
35
docker/gpu.Dockerfile
Normal file
35
docker/gpu.Dockerfile
Normal file
@@ -0,0 +1,35 @@
|
||||
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"]
|
||||
Reference in New Issue
Block a user