Skip to content

Commit 4d120db

Browse files
committed
fixed error of ext build
1 parent e9bb120 commit 4d120db

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

python-kaldi/Dockerfile.py38-cuda11.1

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,30 +3,31 @@ FROM lablup/common-base:py38-cuda11.1
33
# install kaldi
44
WORKDIR /opt
55
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 && \
6+
RUN git clone https://github.com/hephaex/kaldi.git /opt/kaldi && \
97
cd /opt/kaldi/tools && \
108
make -j $(nproc) && \
119
cd /opt/kaldi/src && \
1210
./configure --shared --use-cuda \
1311
--cudatk-dir=/usr/local/cuda/ \
1412
--mathlib=ATLAS \
1513
--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" && \
16-
make depend -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs && \
17-
make -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs
14+
make depend -j $(nproc) && \
15+
make -j $(nproc)
1816
RUN cd /opt/kaldi/src && \
19-
make ext -j $(nproc) -lcuda -L /usr/local/cuda/lib64/stubs && \
17+
make ext -j $(nproc) && \
2018
ldconfig && \
2119
find /opt/kaldi -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -\
2220
exec rm {} \; && \
2321
rm -rf /opt/kaldi/.git
2422

23+
RUN mkdir -p /workspace && \
24+
chmod -R 777 /workspace
2525
# Install ipython kernelspec
2626
Run python3 -m ipykernel install --display-name "Lablup Kaldi 21.01 on Python 3\
2727
.8 (CUDA 11.1)" && \
2828
cat /usr/local/share/jupyter/kernels/python3/kernel.json
2929

30+
3031
# Backend.AI specifics
3132
LABEL ai.backend.kernelspec="1" \
3233
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

0 commit comments

Comments
 (0)