From 55f2c41ec661239c5eeaea8d6754eca0ff9d7f7b Mon Sep 17 00:00:00 2001
From: Bernard <bernard.scumm@kakaobrain.com>
Date: Tue, 2 Feb 2021 01:39:48 +0900
Subject: [PATCH] Support dockerfile

---
 Dockerfile | 72 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..936964a
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,72 @@
+FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
+
+WORKDIR /app
+
+RUN apt-get update && \
+    apt-get install -y apt-utils \
+    wget \
+    git \
+    gcc \
+    build-essential \
+    cmake \
+    libpq-dev \
+    libsndfile-dev \
+    libboost-system-dev \
+    libboost-thread-dev \
+    libboost-program-options-dev \
+    libboost-test-dev \
+    libeigen3-dev \
+    zlib1g-dev \
+    libbz2-dev \
+    liblzma-dev \
+    libsndfile1-dev \
+    libopenblas-dev \
+    libfftw3-dev \
+    libgflags-dev \
+    libgoogle-glog-dev \
+    libgl1-mesa-glx \
+    libomp-dev
+
+# 1. upgrade pip
+RUN pip install --upgrade pip
+
+# 2. install python-mecab-ko
+RUN pip install python-mecab-ko==1.0.9
+
+# 3. install pororo
+RUN pip install pororo
+
+# 4. install brainspeech
+RUN pip install soundfile \
+    torchaudio==0.6.0 \
+    pydub
+
+RUN conda install -y -c conda-forge librosa
+
+# 5. install etc modules
+RUN pip install librosa \
+    kollocate \
+    koparadigm \
+    g2pk \
+    fugashi \
+    ipadic \
+    romkan \
+    g2pM \
+    jieba \
+    opencv-python \
+    scikit-image
+
+WORKDIR /app/external_lib
+
+RUN git clone https://github.com/kpu/kenlm.git
+WORKDIR /app/external_lib/kenlm/build
+RUN cmake .. -DCMAKE_BUILD_TYPE=Release -DCMAKE_POSITION_INDEPENDENT_CODE=ON
+RUN make
+ENV KENLM_ROOT_DIR="/app/external_lib/kenlm/"
+
+WORKDIR /app/external_lib
+RUN git clone -b v0.2 https://github.com/facebookresearch/wav2letter.git
+WORKDIR /app/external_lib/wav2letter/bindings/python
+RUN pip install -e .
+
+WORKDIR /app