-
Notifications
You must be signed in to change notification settings - Fork 295
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add CUDA docker image for torch 2.2.1 (#1521)
- Loading branch information
1 parent
7e2b561
commit 2f102eb
Showing
13 changed files
with
160 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| FROM pytorch/pytorch:2.2.1-cuda11.8-cudnn8-devel | ||
|
|
||
| ENV LC_ALL C.UTF-8 | ||
|
|
||
| ARG DEBIAN_FRONTEND=noninteractive | ||
|
|
||
| # python 3.10 | ||
| ARG K2_VERSION="1.24.4.dev20240223+cuda11.8.torch2.2.1" | ||
| ARG KALDIFEAT_VERSION="1.25.4.dev20240223+cuda11.8.torch2.2.1" | ||
| ARG TORCHAUDIO_VERSION="2.2.1+cu118" | ||
|
|
||
| LABEL authors="Fangjun Kuang <csukuangfj@gmail.com>" | ||
| LABEL k2_version=${K2_VERSION} | ||
| LABEL kaldifeat_version=${KALDIFEAT_VERSION} | ||
| LABEL github_repo="https://github.com/k2-fsa/icefall" | ||
|
|
||
| RUN apt-get update && \ | ||
| apt-get install -y --no-install-recommends \ | ||
| curl \ | ||
| vim \ | ||
| libssl-dev \ | ||
| autoconf \ | ||
| automake \ | ||
| bzip2 \ | ||
| ca-certificates \ | ||
| ffmpeg \ | ||
| g++ \ | ||
| gfortran \ | ||
| git \ | ||
| libtool \ | ||
| make \ | ||
| patch \ | ||
| sox \ | ||
| subversion \ | ||
| unzip \ | ||
| valgrind \ | ||
| wget \ | ||
| zlib1g-dev \ | ||
| && rm -rf /var/lib/apt/lists/* | ||
|
|
||
| # Install dependencies | ||
| RUN pip install --no-cache-dir \ | ||
| torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/torch_stable.html \ | ||
| k2==${K2_VERSION} -f https://k2-fsa.github.io/k2/cuda.html \ | ||
| git+https://github.com/lhotse-speech/lhotse \ | ||
| kaldifeat==${KALDIFEAT_VERSION} -f https://csukuangfj.github.io/kaldifeat/cuda.html \ | ||
| kaldi_native_io \ | ||
| kaldialign \ | ||
| kaldifst \ | ||
| kaldilm \ | ||
| sentencepiece>=0.1.96 \ | ||
| tensorboard \ | ||
| typeguard \ | ||
| dill \ | ||
| onnx \ | ||
| onnxruntime \ | ||
| onnxmltools \ | ||
| multi_quantization \ | ||
| typeguard \ | ||
| numpy \ | ||
| pytest \ | ||
| graphviz | ||
|
|
||
| RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \ | ||
| cd /workspace/icefall && \ | ||
| pip install --no-cache-dir -r requirements.txt | ||
|
|
||
| ENV PYTHONPATH /workspace/icefall:$PYTHONPATH | ||
|
|
||
| WORKDIR /workspace/icefall |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-devel | ||
|
|
||
| ENV LC_ALL C.UTF-8 | ||
|
|
||
| ARG DEBIAN_FRONTEND=noninteractive | ||
|
|
||
| # python 3.10 | ||
| ARG K2_VERSION="1.24.4.dev20240223+cuda12.1.torch2.2.1" | ||
| ARG KALDIFEAT_VERSION="1.25.4.dev20240223+cuda12.1.torch2.2.1" | ||
| ARG TORCHAUDIO_VERSION="2.2.1+cu121" | ||
|
|
||
| LABEL authors="Fangjun Kuang <csukuangfj@gmail.com>" | ||
| LABEL k2_version=${K2_VERSION} | ||
| LABEL kaldifeat_version=${KALDIFEAT_VERSION} | ||
| LABEL github_repo="https://github.com/k2-fsa/icefall" | ||
|
|
||
| RUN apt-get update && \ | ||
| apt-get install -y --no-install-recommends \ | ||
| curl \ | ||
| vim \ | ||
| libssl-dev \ | ||
| autoconf \ | ||
| automake \ | ||
| bzip2 \ | ||
| ca-certificates \ | ||
| ffmpeg \ | ||
| g++ \ | ||
| gfortran \ | ||
| git \ | ||
| libtool \ | ||
| make \ | ||
| patch \ | ||
| sox \ | ||
| subversion \ | ||
| unzip \ | ||
| valgrind \ | ||
| wget \ | ||
| zlib1g-dev \ | ||
| && rm -rf /var/lib/apt/lists/* | ||
|
|
||
| # Install dependencies | ||
| RUN pip install --no-cache-dir \ | ||
| torchaudio==${TORCHAUDIO_VERSION} -f https://download.pytorch.org/whl/torch_stable.html \ | ||
| k2==${K2_VERSION} -f https://k2-fsa.github.io/k2/cuda.html \ | ||
| git+https://github.com/lhotse-speech/lhotse \ | ||
| kaldifeat==${KALDIFEAT_VERSION} -f https://csukuangfj.github.io/kaldifeat/cuda.html \ | ||
| kaldi_native_io \ | ||
| kaldialign \ | ||
| kaldifst \ | ||
| kaldilm \ | ||
| sentencepiece>=0.1.96 \ | ||
| tensorboard \ | ||
| typeguard \ | ||
| dill \ | ||
| onnx \ | ||
| onnxruntime \ | ||
| onnxmltools \ | ||
| multi_quantization \ | ||
| typeguard \ | ||
| numpy \ | ||
| pytest \ | ||
| graphviz | ||
|
|
||
| RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \ | ||
| cd /workspace/icefall && \ | ||
| pip install --no-cache-dir -r requirements.txt | ||
|
|
||
| ENV PYTHONPATH /workspace/icefall:$PYTHONPATH | ||
|
|
||
| WORKDIR /workspace/icefall |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters