From d30244e28f474f53a31556d49008bd8ef1278ed3 Mon Sep 17 00:00:00 2001
From: cdxie <xcdahu@mail.ustc.edu.cn>
Date: Tue, 19 Oct 2021 13:00:59 +0800
Subject: [PATCH] add a docker file for some users (#87)

* add a docker file for some users

Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8-python3.8

* add a describing file of how to use dockerfile

give some steps to use dockerfile
---
 docker/README.md                              | 24 +++++
 .../Dockerfile                                | 91 +++++++++++++++++++
 2 files changed, 115 insertions(+)
 create mode 100644 docker/README.md
 create mode 100644 docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8/Dockerfile

diff --git a/docker/README.md b/docker/README.md
new file mode 100644
index 000000000..0c8cb0ed9
--- /dev/null
+++ b/docker/README.md
@@ -0,0 +1,24 @@
+# icefall dockerfile
+
+We provide a dockerfile for some users, the configuration of dockerfile is : Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8-python3.8. You can use the dockerfile by following the steps:
+
+## Building images locally
+
+```bash
+cd docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8
+docker build -t icefall/pytorch1.7.1:latest -f ./Dockerfile ./
+```
+
+## Using built images 
+Sample usage of the GPU based images:
+Note: use [nvidia-docker](https://github.com/NVIDIA/nvidia-docker) to run the GPU images.
+
+```bash
+docker run -it --runtime=nvidia --name=icefall_username --gpus all icefall/pytorch1.7.1:latest
+```
+
+Sample usage of the CPU based images:
+
+```bash
+docker run -it icefall/pytorch1.7.1:latest /bin/bash
+``` 
\ No newline at end of file
diff --git a/docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8/Dockerfile b/docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8/Dockerfile
new file mode 100644
index 000000000..a9caf07ed
--- /dev/null
+++ b/docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8/Dockerfile
@@ -0,0 +1,91 @@
+FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
+
+# install normal source
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        g++ \
+        make \
+        automake \
+        autoconf \
+        bzip2 \
+        unzip \
+        wget \
+        sox \
+        libtool \
+        git \
+        subversion \
+        zlib1g-dev \
+        gfortran \
+        ca-certificates \
+        patch \
+        ffmpeg \
+        valgrind \
+	libssl-dev \
+	    vim && \
+        rm -rf /var/lib/apt/lists/*
+
+
+RUN mv /opt/conda/lib/libcufft.so.10 /opt/libcufft.so.10.bak && \
+    mv /opt/conda/lib/libcurand.so.10 /opt/libcurand.so.10.bak && \
+    mv /opt/conda/lib/libcublas.so.11 /opt/libcublas.so.11.bak && \
+    mv /opt/conda/lib/libnvrtc.so.11.0 /opt/libnvrtc.so.11.1.bak && \
+    mv /opt/conda/lib/libnvToolsExt.so.1 /opt/libnvToolsExt.so.1.bak && \
+    mv /opt/conda/lib/libcudart.so.11.0 /opt/libcudart.so.11.0.bak
+
+# cmake
+
+RUN wget -P /opt https://cmake.org/files/v3.18/cmake-3.18.0.tar.gz && \
+    cd /opt && \
+    tar -zxvf cmake-3.18.0.tar.gz && \
+    cd cmake-3.18.0 && \
+    ./bootstrap && \
+    make && \
+    make install && \
+    rm -rf cmake-3.18.0.tar.gz && \
+    find /opt/cmake-3.18.0 -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \
+    cd -
+
+#kaldiio
+
+RUN pip install kaldiio
+
+# flac 
+RUN wget -P /opt https://downloads.xiph.org/releases/flac/flac-1.3.2.tar.xz  && \
+    cd /opt && \ 
+    xz -d flac-1.3.2.tar.xz && \
+    tar -xvf flac-1.3.2.tar && \
+    cd flac-1.3.2 && \
+    ./configure && \
+    make && make install && \
+    rm -rf flac-1.3.2.tar && \
+    find /opt/flac-1.3.2  -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \
+    cd - 
+
+# graphviz
+RUN pip install graphviz
+
+# kaldifeat
+RUN git clone https://github.com/csukuangfj/kaldifeat.git /opt/kaldifeat && \
+    cd /opt/kaldifeat && \
+    python setup.py install && \
+    cd -
+
+
+#install k2 from source
+RUN git clone https://github.com/k2-fsa/k2.git /opt/k2 && \
+    cd /opt/k2 && \
+    python3 setup.py install && \
+    cd -
+
+# install  lhotse
+RUN pip install git+https://github.com/lhotse-speech/lhotse
+#RUN pip install lhotse
+
+# install icefall
+RUN git clone https://github.com/k2-fsa/icefall && \
+    cd icefall && \
+    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
+    
+ENV PYTHONPATH /workspace/icefall:$PYTHONPATH  
+