Fix installing torch and torchaudio

k2-fsa · Oct 29, 2024 · 091136d · 091136d
1 parent 56523fd
commit 091136d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/.github/scripts/docker/Dockerfile b/.github/scripts/docker/Dockerfile
@@ -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 \

diff --git a/.github/scripts/docker/generate_build_matrix.py b/.github/scripts/docker/generate_build_matrix.py
@@ -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"]

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