From aa195675c0473fae2092fd363eea6c7447cbb5c3 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Mon, 12 Dec 2022 19:28:07 -0500
Subject: [PATCH] Dockerfile for softvc

---
 voice_conversion/softvc/.dockerignore | 16 +++++++++
 voice_conversion/softvc/Dockerfile    | 51 +++++++++++++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 voice_conversion/softvc/.dockerignore
 create mode 100644 voice_conversion/softvc/Dockerfile

diff --git a/voice_conversion/softvc/.dockerignore b/voice_conversion/softvc/.dockerignore
new file mode 100644
index 0000000..da18c05
--- /dev/null
+++ b/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
+
diff --git a/voice_conversion/softvc/Dockerfile b/voice_conversion/softvc/Dockerfile
new file mode 100644
index 0000000..83c52c1
--- /dev/null
+++ b/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 . ./