mirror of https://github.com/coqui-ai/TTS.git
45 lines
1.0 KiB
Docker
45 lines
1.0 KiB
Docker
ARG BASE=nvidia/cuda:11.8.0-base-ubuntu22.04
|
|
FROM ${BASE}
|
|
|
|
# Install OS dependencies:
|
|
RUN apt-get update && apt-get upgrade -y
|
|
RUN apt-get install -y --no-install-recommends \
|
|
gcc g++ \
|
|
make \
|
|
python3 python3-dev python3-pip python3-venv python3-wheel \
|
|
espeak-ng libsndfile1-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install Major Python Dependencies:
|
|
RUN pip3 install llvmlite --ignore-installed
|
|
RUN pip3 install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
|
|
RUN rm -rf /root/.cache/pip
|
|
|
|
WORKDIR /root
|
|
|
|
# Copy Dependency Lock Files:
|
|
COPY \
|
|
Makefile \
|
|
pyproject.toml \
|
|
setup.py \
|
|
requirements.dev.txt \
|
|
requirements.ja.txt \
|
|
requirements.notebooks.txt \
|
|
requirements.txt \
|
|
/root/
|
|
|
|
# Install Project Dependencies
|
|
# Separate stage to limit re-downloading:
|
|
RUN pip install \
|
|
-r requirements.txt \
|
|
-r requirements.dev.txt \
|
|
-r requirements.ja.txt \
|
|
-r requirements.notebooks.txt
|
|
|
|
# Copy TTS repository contents:
|
|
COPY . /root
|
|
|
|
# Installing the TTS package itself:
|
|
RUN make install
|
|
|