mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-10 02:22:17 +00:00
67 lines
2.0 KiB
YAML
67 lines
2.0 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.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 }}
|
|
run: |
|
|
uname -a
|
|
cat /etc/*release
|
|
|
|
nvcc --version
|
|
|
|
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
|
|
|
|
pushd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && popd
|
|
|
|
export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH
|
|
echo "LD_LIBRARY_PATH $LD_LIBRARY_PATH"
|
|
|
|
python3 --version
|
|
which python3
|
|
|
|
echo "----------torch----------"
|
|
python3 -m torch.utils.collect_env
|
|
|
|
echo "----------k2----------"
|
|
python3 -c "import k2; print(k2.__file__)"
|
|
python3 -c "import k2; print(k2.__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__)"
|
|
|