Merge branch 'master' of https://github.com/lablup/backend.ai-kernels

Author: hephaex

build.py

@@ -78,6 +78,7 @@ def build_common(name, tag, extra_opts=''):
     'tf-old', 'tf-current', 'tf-future',
     'caffe', 'pytorch',
     'cntk-builder','cntk',
+    'kaldi',
     'vendor-aws', 'vendor-ngc', 'vendor-freecad',
 ]
 
@@ -338,6 +339,9 @@ def main(build, list_builds, _auto_push):
         build_kernel('python-cntk', '2.7-py36')
         build_kernel('python-cntk', '2.7-py36-cuda9')
 
+    if 'kaldi' in build:
+        build_kernel('python-kaldi', 'py38-cuda11.1')
+
     # AWS polly
     if 'vendor-aws' in build:
         build_kernel('vendor/aws_polly', '0.1-alpine3.8')

python-kaldi/Dockerfile.py38-cuda11.1

@@ -3,30 +3,31 @@ FROM lablup/common-base:py38-cuda11.1
 # install kaldi
 WORKDIR /opt
 RUN apt update ; apt-get install -y sox subversion
-RUN git clone https://github.com/kaldi-asr/kaldi.git /opt/kaldi && \
-    cd /opt/kaldi && \
-    git checkout c013e2e && \
+RUN git clone https://github.com/hephaex/kaldi.git /opt/kaldi && \
     cd /opt/kaldi/tools && \
     make -j $(nproc) && \
     cd /opt/kaldi/src && \
     ./configure --shared --use-cuda \
 	--cudatk-dir=/usr/local/cuda/ \
 	--mathlib=ATLAS \
     	--cuda-arch="-gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80" && \
-    make depend -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs  && \
-    make -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs
+    make depend -j $(nproc) && \
+    make -j $(nproc) 
 RUN cd /opt/kaldi/src  && \
-    make ext -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs  && \
+    make ext -j $(nproc) && \
     ldconfig && \
     find /opt/kaldi  -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -\
 exec rm {} \; && \
     rm -rf /opt/kaldi/.git
 
+RUN mkdir -p /workspace && \
+    chmod -R 777 /workspace
 # Install ipython kernelspec
 Run python3 -m ipykernel install --display-name "Lablup Kaldi 21.01 on Python 3\
 .8 (CUDA 11.1)" && \
     cat /usr/local/share/jupyter/kernels/python3/kernel.json
 
+
 # Backend.AI specifics
 LABEL ai.backend.kernelspec="1" \
       ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \