|
|
FROM pytorch/pytorch:2.2.2-cuda11.8-cudnn8-devel |
|
|
|
|
|
|
|
|
ENV LC_ALL C.UTF-8 |
|
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive |
|
|
|
|
|
|
|
|
ARG K2_VERSION="1.24.4.dev20240328+cuda11.8.torch2.2.2" |
|
|
ARG KALDIFEAT_VERSION="1.25.4.dev20240329+cuda11.8.torch2.2.2" |
|
|
ARG TORCHAUDIO_VERSION="2.2.2+cu118" |
|
|
|
|
|
LABEL authors="Fangjun Kuang <csukuangfj@gmail.com>" |
|
|
LABEL k2_version=${K2_VERSION} |
|
|
LABEL kaldifeat_version=${KALDIFEAT_VERSION} |
|
|
LABEL github_repo="https://github.com/k2-fsa/icefall" |
|
|
|
|
|
RUN apt-get update && \ |
|
|
apt-get install -y --no-install-recommends \ |
|
|
curl \ |
|
|
vim \ |
|
|
libssl-dev \ |
|
|
autoconf \ |
|
|
automake \ |
|
|
bzip2 \ |
|
|
ca-certificates \ |
|
|
ffmpeg \ |
|
|
g++ \ |
|
|
gfortran \ |
|
|
git \ |
|
|
libtool \ |
|
|
make \ |
|
|
patch \ |
|
|
sox \ |
|
|
subversion \ |
|
|
unzip \ |
|
|
valgrind \ |
|
|
wget \ |
|
|
zlib1g-dev \ |
|
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
|
RUN pip install --no-cache-dir \ |
|
|
torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/torchaudio/ \ |
|
|
k2==${K2_VERSION} -f https://k2-fsa.github.io/k2/cuda.html \ |
|
|
git+https://github.com/lhotse-speech/lhotse \ |
|
|
kaldifeat==${KALDIFEAT_VERSION} -f https://csukuangfj.github.io/kaldifeat/cuda.html \ |
|
|
kaldi_native_io \ |
|
|
kaldialign \ |
|
|
kaldifst \ |
|
|
kaldilm \ |
|
|
sentencepiece>=0.1.96 \ |
|
|
tensorboard \ |
|
|
typeguard \ |
|
|
dill \ |
|
|
onnx \ |
|
|
onnxruntime \ |
|
|
onnxmltools \ |
|
|
onnxoptimizer \ |
|
|
onnxsim \ |
|
|
multi_quantization \ |
|
|
typeguard \ |
|
|
numpy \ |
|
|
pytest \ |
|
|
graphviz |
|
|
|
|
|
RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \ |
|
|
cd /workspace/icefall && \ |
|
|
pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
ENV PYTHONPATH /workspace/icefall:$PYTHONPATH |
|
|
|
|
|
WORKDIR /workspace/icefall |
|
|
|
