mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 04:22:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix installing torch and torchaudio

Browse Source
This commit is contained in:
Fangjun Kuang 2024-10-29 21:33:49 +08:00
parent 56523fd7eb
commit 091136db76
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/scripts/docker/Dockerfile vendored
View File

@ -31,7 +31,8 @@ LABEL github_repo="https://github.com/k2-fsa/icefall"
# Install dependencies # Install dependencies
RUN pip install --no-cache-dir \ RUN pip install --no-cache-dir \
torch==${TORCH_VERSION} torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/cpu/torch_stable.html \ torch==${TORCH_VERSION}+cpu -f https://download.pytorch.org/whl/torch \
torchaudio==${TORCHAUDIO_VERSION}+cpu -f https://download.pytorch.org/whl/torchaudio \
k2==${_K2_VERSION} -f https://k2-fsa.github.io/k2/cpu.html \ k2==${_K2_VERSION} -f https://k2-fsa.github.io/k2/cpu.html \
\ \
git+https://github.com/lhotse-speech/lhotse \ git+https://github.com/lhotse-speech/lhotse \
@ -51,7 +52,7 @@ RUN pip install --no-cache-dir \
matplotlib \ matplotlib \
multi_quantization \ multi_quantization \
numba \ numba \
numpy \ "numpy<2.0" \
onnxoptimizer \ onnxoptimizer \
onnxsim \ onnxsim \
onnx \ onnx \

4
.github/scripts/docker/generate_build_matrix.py vendored
View File

@ -46,7 +46,9 @@ def get_matrix():
k2_version = "1.24.4.dev20241029" k2_version = "1.24.4.dev20241029"
kaldifeat_version = "1.25.5.dev20241029" kaldifeat_version = "1.25.5.dev20241029"
version = "20241029" version = "20241029"
python_version = ["3.8", "3.9", "3.10", "3.11", "3.12", "3.13"]
# torchaudio 2.5.0 does not support python 3.13
python_version = ["3.8", "3.9", "3.10", "3.11", "3.12"]
torch_version = [] torch_version = []
# torch_version += ["1.13.0", "1.13.1"] # torch_version += ["1.13.0", "1.13.1"]
# torch_version += ["2.0.0", "2.0.1"] # torch_version += ["2.0.0", "2.0.1"]

1
.github/workflows/yesno.yml vendored
View File

@ -61,5 +61,6 @@ jobs:
python3 -m torch.utils.collect_env python3 -m torch.utils.collect_env
python3 -m k2.version python3 -m k2.version
pip list
.github/scripts/yesno/ASR/run.sh .github/scripts/yesno/ASR/run.sh