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

small fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2023-07-28 13:31:53 +08:00
parent 665de11b17
commit d73d8acf24
2 changed files with 19 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
.github/workflows/run-docker-image.yml vendored
View File

@ -38,6 +38,12 @@ jobs:
# For torch1.9.0-cuda10.2 # For torch1.9.0-cuda10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/compat:$LD_LIBRARY_PATH 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
which nvcc which nvcc
cuda_dir=$(dirname $(which nvcc)) cuda_dir=$(dirname $(which nvcc))
@ -48,10 +54,12 @@ jobs:
find / -name libcuda.so* 2>/dev/null find / -name libcuda.so* 2>/dev/null
pushd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && popd # for torch1.13.0-cuda11.6
cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && cd -
export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH
echo "LD_LIBRARY_PATH $LD_LIBRARY_PATH"
find / -name libcuda.so* 2>/dev/null
echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"
python3 --version python3 --version
which python3 which python3

10
docker/torch1.9.0-cuda10.2.dockerfile
View File

@ -15,8 +15,10 @@ LABEL github_repo="https://github.com/k2-fsa/icefall"
# see https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/ # see https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/
RUN apt-key del 7fa2af80 && \ RUN rm /etc/apt/sources.list.d/cuda.list && \
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb | dpkg -i - rm /etc/apt/sources.list.d/nvidia-ml.list && \
apt-key del 7fa2af80
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y --no-install-recommends \ apt-get install -y --no-install-recommends \
@ -42,6 +44,10 @@ RUN apt-get update && \
zlib1g-dev \ zlib1g-dev \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb | dpkg -i - && \
apt-get update && \
rm -rf /var/lib/apt/lists/*
# Install dependencies # Install dependencies
RUN pip install --no-cache-dir \ RUN pip install --no-cache-dir \
torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/torch_stable.html \ torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/torch_stable.html \