mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-30 12:22:58 +00:00
refactor: docker layout
This commit is contained in:
22
docker/conda-gpu.dockerfile
Normal file
22
docker/conda-gpu.dockerfile
Normal file
@@ -0,0 +1,22 @@
|
||||
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.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
|
||||
|
||||
RUN conda install -y cudatoolkit=9.0 \
|
||||
&& conda install -y tensorflow-gpu==1.14.0 \
|
||||
&& 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=1.4.4
|
||||
|
||||
ENTRYPOINT ["spleeter"]
|
||||
Reference in New Issue
Block a user