mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 09:32:20 +00:00
106 lines
3.2 KiB
YAML
106 lines
3.2 KiB
YAML
name: Run docker image
|
|
on:
|
|
workflow_dispatch:
|
|
|
|
concurrency:
|
|
group: run_docker_image-${{ github.ref }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
run-docker-image:
|
|
name: ${{ matrix.image }}
|
|
runs-on: ${{ matrix.os }}
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
os: [ubuntu-latest]
|
|
image: ["torch2.1.0-cuda12.1", "torch2.1.0-cuda11.8", "torch2.0.0-cuda11.7", "torch1.13.0-cuda11.6", "torch1.12.1-cuda11.3", "torch1.9.0-cuda10.2"]
|
|
steps:
|
|
# refer to https://github.com/actions/checkout
|
|
- uses: actions/checkout@v2
|
|
with:
|
|
fetch-depth: 0
|
|
|
|
- name: Run the build process with Docker
|
|
uses: addnab/docker-run-action@v3
|
|
with:
|
|
image: k2fsa/icefall:${{ matrix.image }}
|
|
shell: bash
|
|
run: |
|
|
uname -a
|
|
cat /etc/*release
|
|
|
|
find / -name libcuda* 2>/dev/null
|
|
|
|
ls -lh /usr/local/
|
|
ls -lh /usr/local/cuda*
|
|
|
|
nvcc --version
|
|
|
|
ls -lh /usr/local/cuda-*/compat/*
|
|
|
|
# For torch1.9.0-cuda10.2
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/compat:$LD_LIBRARY_PATH
|
|
|
|
# For torch1.12.1-cuda11.3
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/compat:$LD_LIBRARY_PATH
|
|
|
|
# For torch2.0.0-cuda11.7
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/compat:$LD_LIBRARY_PATH
|
|
|
|
# For torch2.1.0-cuda11.8
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/compat:$LD_LIBRARY_PATH
|
|
|
|
# For torch2.1.0-cuda12.1
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/compat:$LD_LIBRARY_PATH
|
|
|
|
|
|
which nvcc
|
|
cuda_dir=$(dirname $(which nvcc))
|
|
echo "cuda_dir: $cuda_dir"
|
|
|
|
find $cuda_dir -name libcuda.so*
|
|
echo "--------------------"
|
|
|
|
find / -name libcuda.so* 2>/dev/null
|
|
|
|
# for torch1.13.0-cuda11.6
|
|
if [ -e /opt/conda/lib/stubs/libcuda.so ]; then
|
|
cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && cd -
|
|
export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH
|
|
fi
|
|
|
|
find / -name libcuda.so* 2>/dev/null
|
|
echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"
|
|
|
|
python3 --version
|
|
which python3
|
|
|
|
python3 -m pip list
|
|
|
|
echo "----------torch----------"
|
|
python3 -m torch.utils.collect_env
|
|
|
|
echo "----------k2----------"
|
|
python3 -c "import k2; print(k2.__file__)"
|
|
python3 -c "import k2; print(k2.__dev_version__)"
|
|
python3 -m k2.version
|
|
|
|
echo "----------lhotse----------"
|
|
python3 -c "import lhotse; print(lhotse.__file__)"
|
|
python3 -c "import lhotse; print(lhotse.__version__)"
|
|
|
|
echo "----------kaldifeat----------"
|
|
python3 -c "import kaldifeat; print(kaldifeat.__file__)"
|
|
python3 -c "import kaldifeat; print(kaldifeat.__version__)"
|
|
|
|
echo "Test yesno recipe"
|
|
|
|
cd egs/yesno/ASR
|
|
|
|
./prepare.sh
|
|
|
|
./tdnn/train.py
|
|
|
|
./tdnn/decode.py
|