File size: 1,202 Bytes
b5b18c8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# Use an official NVIDIA base image with CUDA support
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
# Set label for the docker image description
LABEL description="Docker image for xtts-api-server"
# Install required packages (avoid cache to reduce image size)
RUN apt-get update && \
apt-get install --no-install-recommends -y \
python3-dev portaudio19-dev libportaudio2 libasound2-dev libportaudiocpp0 \
git python3 python3-pip make g++ ffmpeg && \
rm -rf /var/lib/apt/lists/*
# Upgrade pip and install virtualenv
RUN python3 -m pip install --upgrade pip setuptools wheel ninja virtualenv
# Copy the application source code to /app directory and change the workdir to /app
# COPY . /app
# WORKDIR /app
# Install Python dependencies
RUN pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu121
RUN pip install deepspeed
RUN pip install xtts-api-server
# Expose the container ports
EXPOSE 8020
# Run xtts_api_server when the container starts
CMD ["bash", "-c", "python3 -m xtts_api_server --listen -p 8020 -t 'http://localhost:8020' -sf 'xtts-server/speakers' -o 'xtts-server/output' -mf 'xtts-server/models' --deepspeed"]
|