35 lines
1.4 KiB
Docker
35 lines
1.4 KiB
Docker
FROM nvidia/cuda:12.9.1-cudnn-devel-ubuntu24.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
LANG=C.UTF-8 \
|
|
LC_ALL=C.UTF-8 \
|
|
PATH="/usr/local/bin:/root/.local/bin:/root/bin:/piper/.venv/bin:${PATH}"
|
|
|
|
RUN apt-get update && apt upgrade -y && apt-get install -y --no-install-recommends \
|
|
bash coreutils vim nano \
|
|
git curl wget ca-certificates gnupg \
|
|
build-essential pkg-config ninja-build cmake \
|
|
python3 python3-pip python3-venv python3-dev \
|
|
ffmpeg espeak-ng \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY --from=ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv
|
|
RUN chmod +x /usr/local/bin/uv
|
|
|
|
WORKDIR /piper
|
|
|
|
RUN git clone --depth 1 --branch v1.3.0 https://github.com/OHF-voice/piper1-gpl.git .
|
|
# RUN if ! [ -d wavs ]; then mkdir wavs; fi && if ! [ -d metadata ]; then mkdir metadata; fi && if ! [ -d checkpoints ]; then mkdir checkpoints; fi && if ! [ -d lightning_logs ]; then mkdir lightning_logs; fi && if ! [ -d cache ]; then mkdir cache; fi
|
|
RUN uv python install 3.13 && uv python pin 3.13 && uv venv .venv --python 3.13
|
|
|
|
RUN uv pip install -e .[train] torch==2.5.1 lightning==2.1.0 numpy "ml_dtypes>=0.5.0" scikit-build \
|
|
--index https://download.pytorch.org/whl/cu124 \
|
|
--index https://pypi.org/simple \
|
|
--index-strategy=unsafe-best-match \
|
|
--compile-bytecode
|
|
|
|
RUN uv run python3 setup.py build_ext --inplace
|
|
RUN . .venv/bin/activate && ./build_monotonic_align.sh
|
|
|
|
ENV UV_OFFLINE=1
|