GPT-SoVITS/Dockerfile
2025-05-01 02:29:57 +01:00

70 lines
1.6 KiB
Docker

ARG CUDA_VERSION=12.4
ARG CUDA_BASE=runtime
FROM nvidia/cuda:${CUDA_VERSION}.1-cudnn-${CUDA_BASE}-ubuntu22.04
LABEL maintainer="XXXXRT"
LABEL version="V4-0501"
LABEL description="Docker image for GPT-SoVITS"
ARG CUDA_VERSION=12.4
ENV CUDA_VERSION=${CUDA_VERSION}
RUN DEBIAN_FRONTEND=noninteractive apt-get update -qq 1>/dev/null && \
apt-get install -y -qq --no-install-recommends \
build-essential \
gcc \
g++ \
wget \
curl \
unzip \
git \
nano \
htop \
procps \
ca-certificates \
locales \
1>/dev/null \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /workspace/GPT-SoVITS
COPY . /workspace/GPT-SoVITS
ARG LITE=false
ENV LITE=${LITE}
ARG WORKFLOW=false
ENV WORKFLOW=${WORKFLOW}
ARG TARGETPLATFORM
ENV TARGETPLATFORM=${TARGETPLATFORM}
ENV HOME="/root"
RUN bash Docker/anaconda_install.sh
ENV PATH="$HOME/anaconda3/bin:$PATH"
SHELL ["/bin/bash", "-c"]
ENV PATH="/usr/local/cuda/bin:$PATH"
ENV CUDA_HOME="/usr/local/cuda"
ENV MAKEFLAGS="-j$(nproc)"
RUN bash Docker/setup.sh
EXPOSE 9871 9872 9873 9874 9880
ENV PYTHONPATH="/workspace/GPT-SoVITS"
CMD ["/bin/bash", "-c", "\
source $HOME/anaconda3/etc/profile.d/conda.sh && \
rm -rf /workspace/GPT-SoVITS/GPT_SoVITS/pretrained_models && \
rm -rf /workspace/GPT-SoVITS/tools/asr/models && \
rm -rf /workspace/GPT-SoVITS/tools/uvr5/uvr5_weights && \
ln -s /workspace/model/pretrained_models /workspace/GPT-SoVITS/GPT_SoVITS/pretrained_models && \
ln -s /workspace/model/models /workspace/GPT-SoVITS/tools/asr/models && \
ln -s /workspace/model/uvr5_weights /workspace/GPT-SoVITS/tools/uvr5/uvr5_weights && \
exec bash"]