Skip to content

Commit 50ddb48

Browse files
committed
Kaldi (Python 3.8 & CUDA11.1
1 parent 4be8e84 commit 50ddb48

File tree

1 file changed

+36
-0
lines changed

1 file changed

+36
-0
lines changed

python-kaldi/Dockerfile.py38-cuda11.1

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
FROM lablup/common-base:py38-cuda11.1
2+
3+
# install kaldi
4+
WORKDIR /opt
5+
RUN apt update ; apt-get install -y sox subversion
6+
RUN git clone https://github.com/kaldi-asr/kaldi.git /opt/kaldi && \
7+
cd /opt/kaldi && \
8+
git checkout c013e2e && \
9+
cd /opt/kaldi/tools && \
10+
make -j $(nproc) && \
11+
cd /opt/kaldi/src && \
12+
./configure --shared --use-cuda && \
13+
make depend -j $(nproc) && \
14+
make -j $(nproc) -lcuda && \
15+
find /opt/kaldi -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \
16+
rm -rf /opt/kaldi/.git
17+
18+
# Install ipython kernelspec
19+
Run python3 -m ipykernel install --display-name "Lablup Kaldi 21.01 on Python 3.8 (CUDA 11.1)" && \
20+
cat /usr/local/share/jupyter/kernels/python3/kernel.json
21+
22+
# Backend.AI specifics
23+
LABEL ai.backend.kernelspec="1" \
24+
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
25+
ai.backend.features="batch query uid-match user-input" \
26+
ai.backend.base-distro="ubuntu16.04" \
27+
ai.backend.resource.min.cpu="1" \
28+
ai.backend.resource.min.mem="1g" \
29+
ai.backend.resource.min.cuda.device=0 \
30+
ai.backend.resource.min.cuda.shares=0 \
31+
ai.backend.runtime-type="python" \
32+
ai.backend.runtime-path="/usr/bin/python3" \
33+
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:preopen:5000,nniboard:preopen:8080"
34+
35+
WORKDIR /home/work
36+
# vim: ft=dockerfile

0 commit comments

Comments
 (0)