forked from flashlight/wav2letter
/
Dockerfile-CUDA
29 lines (26 loc) · 1.42 KB
/
Dockerfile-CUDA
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# ==================================================================
# module list
# ------------------------------------------------------------------
# flashlight master (git, CUDA backend)
# ==================================================================
FROM wav2letter/wav2letter:cuda-base-latest
RUN mkdir /root/wav2letter
COPY . /root/wav2letter
# ==================================================================
# flashlight https://github.com/facebookresearch/flashlight.git
# ------------------------------------------------------------------
RUN cd /root && git clone --recursive https://github.com/facebookresearch/flashlight.git && \
cd /root/flashlight && mkdir -p build && \
cd build && cmake .. -DCMAKE_BUILD_TYPE=Release -DFLASHLIGHT_BACKEND=CUDA && \
make -j8 && make install && \
# ==================================================================
# wav2letter with GPU backend
# ------------------------------------------------------------------
export MKLROOT=/opt/intel/mkl && export KENLM_ROOT_DIR=/root/kenlm && \
cd /root/wav2letter && mkdir -p build && \
cd build && cmake .. -DCMAKE_BUILD_TYPE=Release -DW2L_LIBRARIES_USE_CUDA=ON && \
make -j8 && \
# ==================================================================
# wav2letter python bindings
# ------------------------------------------------------------------
cd /root/wav2letter/bindings/python && pip install -e .