Dockerfile for softvc

voice_conversion/softvc/.dockerignore

@@ -0,0 +1,16 @@
+# Python Virtual Env
+python/
+
+# Python object cache
+__pycache__/
+
+# Vim cruft
+*.swo
+*.swp
+*~
+
+# Default output files
+output.wav
+temp.wav
+metadata.json
+

voice_conversion/softvc/Dockerfile

@@ -0,0 +1,51 @@
+
+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 . ./