|
|
@@ -0,0 +1,50 @@
|
|
|
+FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
|
|
|
+
|
|
|
+ENV DEBIAN_FRONTEND=noninteractive
|
|
|
+ENV PYTHONUNBUFFERED=1
|
|
|
+ENV COQUI_TTS_CACHE=/models
|
|
|
+ENV TORCH_HOME=/models
|
|
|
+
|
|
|
+WORKDIR /app
|
|
|
+
|
|
|
+# Ubuntu repositories
|
|
|
+RUN printf 'deb http://archive.ubuntu.com/ubuntu jammy main universe\n\
|
|
|
+deb http://archive.ubuntu.com/ubuntu jammy-updates main universe\n\
|
|
|
+deb http://archive.ubuntu.com/ubuntu jammy-security main universe\n' > /etc/apt/sources.list
|
|
|
+
|
|
|
+# System dependencies
|
|
|
+RUN apt-get update && apt-get install -y \
|
|
|
+ python3 \
|
|
|
+ python3-pip \
|
|
|
+ git \
|
|
|
+ ffmpeg \
|
|
|
+ espeak-ng \
|
|
|
+ libsndfile1 \
|
|
|
+ && rm -rf /var/lib/apt/lists/*
|
|
|
+
|
|
|
+# Upgrade pip
|
|
|
+RUN pip3 install --no-cache-dir --upgrade pip
|
|
|
+
|
|
|
+# CUDA-enabled PyTorch
|
|
|
+RUN pip3 install --no-cache-dir \
|
|
|
+ torch \
|
|
|
+ torchvision \
|
|
|
+ torchaudio \
|
|
|
+ --index-url https://download.pytorch.org/whl/cu118
|
|
|
+
|
|
|
+# Install Coqui TTS + server
|
|
|
+RUN pip3 install --no-cache-dir \
|
|
|
+ TTS[all] \
|
|
|
+ fastapi \
|
|
|
+ uvicorn
|
|
|
+
|
|
|
+# Create directories
|
|
|
+RUN mkdir -p /models /voices
|
|
|
+
|
|
|
+# Ports
|
|
|
+EXPOSE 5002
|
|
|
+
|
|
|
+# Optional volume mounts
|
|
|
+VOLUME ["/models", "/voices"]
|
|
|
+
|
|
|
+ENTRYPOINT ["python3", "-m", "uvicorn", "tts_server:app", "--host", "0.0.0.0", "--port", "5002"]
|