mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-16 00:13:07 +00:00
Brandon Thomas
52 lines
1.9 KiB
Docker
52 lines
1.9 KiB
Docker
|
|
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
|
|
|
|
# See: https://github.com/opencontainers/image-spec/blob/master/annotations.md
|
|
LABEL org.opencontainers.image.title='Storyteller ML: Voice Conversion'
|
|
LABEL org.opencontainers.image.authors='bt@brand.io, echelon@gmail.com'
|
|
LABEL org.opencontainers.image.description='From the Storyteller ML model monorepo: TTS, etc.'
|
|
LABEL org.opencontainers.image.documentation='https://github.com/storytold/tts-core'
|
|
LABEL org.opencontainers.image.source='https://github.com/storytold/tts-core'
|
|
LABEL org.opencontainers.image.url='https://github.com/storytold/tts-core'
|
|
|
|
# tzdata install hangs without presetting the time zone
|
|
ENV TZ=UTC
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
|
|
RUN apt-get update \
|
|
&& DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y \
|
|
curl \
|
|
ffmpeg \
|
|
libmagic-dev \
|
|
libmagic1 \
|
|
python3-pip \
|
|
python3.10 \
|
|
python3.10-venv \
|
|
unzip \
|
|
wget \
|
|
--no-install-recommends
|
|
|
|
# NB: Emit version for build debugging
|
|
RUN python3 --version
|
|
|
|
COPY requirements.txt .
|
|
|
|
# # NB: Forget about virtual environments inside the container.
|
|
# # Also note the --extra-index-url, which is required for pytorch.
|
|
# # The version persisted to 'requirements.txt' chokes without it.
|
|
# RUN pip install --upgrade pip \
|
|
# && pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 \
|
|
# && pip install -r requirements.txt
|
|
|
|
# NB: Install requirements in a virtual env too.
|
|
# Also note the --extra-index-url, which is required for pytorch.
|
|
# The version persisted to 'requirements.txt' chokes without it.
|
|
RUN python3 -m venv python
|
|
RUN . python/bin/activate \
|
|
&& pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 \
|
|
&& pip install -r requirements.txt \
|
|
&& deactivate
|
|
|
|
# Copy remaining files
|
|
COPY . ./
|