mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-12 11:32:19 +00:00
Fix installing torch and torchaudio
This commit is contained in:
parent
56523fd7eb
commit
091136db76
5
.github/scripts/docker/Dockerfile
vendored
5
.github/scripts/docker/Dockerfile
vendored
@ -31,7 +31,8 @@ LABEL github_repo="https://github.com/k2-fsa/icefall"
|
||||
|
||||
# Install dependencies
|
||||
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 \
|
||||
\
|
||||
git+https://github.com/lhotse-speech/lhotse \
|
||||
@ -51,7 +52,7 @@ RUN pip install --no-cache-dir \
|
||||
matplotlib \
|
||||
multi_quantization \
|
||||
numba \
|
||||
numpy \
|
||||
"numpy<2.0" \
|
||||
onnxoptimizer \
|
||||
onnxsim \
|
||||
onnx \
|
||||
|
@ -46,7 +46,9 @@ def get_matrix():
|
||||
k2_version = "1.24.4.dev20241029"
|
||||
kaldifeat_version = "1.25.5.dev20241029"
|
||||
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 += ["1.13.0", "1.13.1"]
|
||||
# torch_version += ["2.0.0", "2.0.1"]
|
||||
|
1
.github/workflows/yesno.yml
vendored
1
.github/workflows/yesno.yml
vendored
@ -61,5 +61,6 @@ jobs:
|
||||
|
||||
python3 -m torch.utils.collect_env
|
||||
python3 -m k2.version
|
||||
pip list
|
||||
|
||||
.github/scripts/yesno/ASR/run.sh
|
||||
|
Loading…
x
Reference in New Issue
Block a user