fix torchaudio version in dockerfile (#653)

* fix torchaudio version in dockerfile

* remove kaldiio
This commit is contained in:
Teo Wen Shen 2022-11-03 11:27:18 +09:00 committed by GitHub
parent 5d285625cf
commit d2a1c65c5c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 6 deletions

View File

@ -72,14 +72,14 @@ docker run -it --runtime=nvidia --shm-size=2gb --name=icefall --gpus all icefall
``` ```
### Tips: ### Tips:
1. Since your data and models most probably won't be in the docker, you must use the -v flag to access the host machine. Do this by specifying `-v {/path/in/docker}:{/path/in/host/machine}`. 1. Since your data and models most probably won't be in the docker, you must use the -v flag to access the host machine. Do this by specifying `-v {/path/in/host/machine}:{/path/in/docker}`.
2. Also, if your environment requires a proxy, this would be a good time to add it in too: `-e http_proxy=http://aaa.bb.cc.net:8080 -e https_proxy=http://aaa.bb.cc.net:8080`. 2. Also, if your environment requires a proxy, this would be a good time to add it in too: `-e http_proxy=http://aaa.bb.cc.net:8080 -e https_proxy=http://aaa.bb.cc.net:8080`.
Overall, your docker run command should look like this. Overall, your docker run command should look like this.
```bash ```bash
docker run -it --runtime=nvidia --shm-size=2gb --name=icefall --gpus all -v {/path/in/docker}:{/path/in/host/machine} -e http_proxy=http://aaa.bb.cc.net:8080 -e https_proxy=http://aaa.bb.cc.net:8080 icefall/pytorch1.12.1 docker run -it --runtime=nvidia --shm-size=2gb --name=icefall --gpus all -v {/path/in/host/machine}:{/path/in/docker} -e http_proxy=http://aaa.bb.cc.net:8080 -e https_proxy=http://aaa.bb.cc.net:8080 icefall/pytorch1.12.1
``` ```
You can explore more docker run options [here](https://docs.docker.com/engine/reference/commandline/run/) to suit your environment. You can explore more docker run options [here](https://docs.docker.com/engine/reference/commandline/run/) to suit your environment.

View File

@ -51,8 +51,9 @@ RUN wget -P /opt https://downloads.xiph.org/releases/flac/flac-1.3.2.tar.xz &&
find /opt/flac-1.3.2 -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \ find /opt/flac-1.3.2 -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \
cd - cd -
RUN pip install kaldiio graphviz && \ RUN conda install -y -c pytorch torchaudio=0.12 && \
conda install -y -c pytorch torchaudio pip install graphviz
#install k2 from source #install k2 from source
RUN git clone https://github.com/k2-fsa/k2.git /opt/k2 && \ RUN git clone https://github.com/k2-fsa/k2.git /opt/k2 && \

View File

@ -69,8 +69,8 @@ RUN wget -P /opt https://downloads.xiph.org/releases/flac/flac-1.3.2.tar.xz &&
find /opt/flac-1.3.2 -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \ find /opt/flac-1.3.2 -type f \( -name "*.o" -o -name "*.la" -o -name "*.a" \) -exec rm {} \; && \
cd - cd -
RUN pip install kaldiio graphviz && \ RUN conda install -y -c pytorch torchaudio=0.7.1 && \
conda install -y -c pytorch torchaudio=0.7.1 pip install graphviz
#install k2 from source #install k2 from source
RUN git clone https://github.com/k2-fsa/k2.git /opt/k2 && \ RUN git clone https://github.com/k2-fsa/k2.git /opt/k2 && \