mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
41 lines
1.9 KiB
Docker
41 lines
1.9 KiB
Docker
FROM python:3.7
|
|
|
|
# Cuda installation.
|
|
# Note : original cuda base file from https://gitlab.com/nvidia/container-images/cuda/blob/master/dist/ubuntu18.04/10.1/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/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 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
ENV CUDA_VERSION 10.1.243
|
|
ENV CUDA_PKG_VERSION 10-1=$CUDA_VERSION-1
|
|
|
|
# For libraries in the cuda-compat-* package: https://docs.nvidia.com/cuda/eula/index.html#attachment-a
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends cuda-cudart-$CUDA_PKG_VERSION \
|
|
&& cuda-compat-10-1 \
|
|
&& ln -s cuda-10.1 /usr/local/cuda \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Required for nvidia-docker v1
|
|
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
|
|
|
|
# nvidia-container-runtime
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
|
ENV NVIDIA_REQUIRE_CUDA "cuda>=10.1 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=396,driver<397 brand=tesla,driver>=410,driver<411"
|
|
|
|
# Spleeter installation.
|
|
RUN apt-get update && apt-get install -y ffmpeg libsndfile1
|
|
RUN pip install musdb museval
|
|
RUN pip install spleeter-gpu==1.4.4
|
|
RUN mkdir -p /model
|
|
ENV MODEL_PATH /model
|
|
ENTRYPOINT ["spleeter"] |