Update docker

This commit is contained in:
Fangjun Kuang 2024-03-18 19:12:44 +08:00
parent 40668e4e8b
commit 557bf292a2
2 changed files with 6 additions and 2 deletions

View File

@ -36,7 +36,9 @@ RUN pip install --no-cache-dir \
\ \
git+https://github.com/lhotse-speech/lhotse \ git+https://github.com/lhotse-speech/lhotse \
kaldifeat==${_KALDIFEAT_VERSION} -f https://csukuangfj.github.io/kaldifeat/cpu.html \ kaldifeat==${_KALDIFEAT_VERSION} -f https://csukuangfj.github.io/kaldifeat/cpu.html \
cython \
dill \ dill \
espnet_tts_frontend \
graphviz \ graphviz \
kaldi-decoder \ kaldi-decoder \
kaldi_native_io \ kaldi_native_io \
@ -45,13 +47,15 @@ RUN pip install --no-cache-dir \
kaldilm \ kaldilm \
matplotlib \ matplotlib \
multi_quantization \ multi_quantization \
numba \
numpy \ numpy \
onnx \ onnx \
onnxmltools \ onnxmltools \
onnxruntime \ onnxruntime \
piper_phonemize -f https://k2-fsa.github.io/icefall/piper_phonemize.html \
pypinyin==0.50.0 \
pytest \ pytest \
sentencepiece>=0.1.96 \ sentencepiece>=0.1.96 \
pypinyin==0.50.0 \
six \ six \
tensorboard \ tensorboard \
typeguard typeguard

View File

@ -45,7 +45,7 @@ def get_torchaudio_version(torch_version):
def get_matrix(): def get_matrix():
k2_version = "1.24.4.dev20240223" k2_version = "1.24.4.dev20240223"
kaldifeat_version = "1.25.4.dev20240223" kaldifeat_version = "1.25.4.dev20240223"
version = "20240223" version = "20240318"
python_version = ["3.8", "3.9", "3.10", "3.11", "3.12"] python_version = ["3.8", "3.9", "3.10", "3.11", "3.12"]
torch_version = ["1.13.0", "1.13.1", "2.0.0", "2.0.1", "2.1.0", "2.1.1", "2.1.2"] torch_version = ["1.13.0", "1.13.1", "2.0.0", "2.0.1", "2.1.0", "2.1.1", "2.1.2"]
torch_version += ["2.2.0", "2.2.1"] torch_version += ["2.2.0", "2.2.1"]