ESPnet安装

版本:cuda 10.1 pytorch1.4 centos7.8

0.准备

需要准备安装好anaconda3,kalid(可以见kaldi安装)

1.创建虚拟环境并激活

执行下面命令创建虚拟环境:

1
2
conda create -n espnet python=3.7.3
conda activate espnet

conda换源,建议换清华的源,阿里的源下载的pytorch和cuda以及cudnn的版本和清华的不同,取决于你的cuda_cudnn的版本,不然会报错
pip换源

参考:https://blog.csdn.net/zhayushui/article/details/80433768

2.安装pytorch

1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1

上面为例子,需要自己去pytorch找对应版本。pytorch官网:https://pytorch.org/

3.克隆espnet

1
2
cd ~
git clone https://github.com/espnet/espnet

我安装在家目录下,自己视情况决定安装的位置
可能github网速挺慢的导致克隆失败,也需要换国内github源

4.在espnet/tools下面链接kaldi

1
2
cd espnet/tools
ln -s ~/kaldi ./

5.设置CUDA环境

1
. ./setup_cuda_env.sh ~/cuda-10.1

注意上面的cuda环境是我自己的环境

6.设置系统python环境

1
./setup_python.sh $(command -v python3)

7.安装

1
make

8.安装warp-ctc

1
2
cd installers
./install_warp-ctc.sh

9.安装warp-transducer

1
./install_warp-transducer.sh

(非必要安装,可以用的时候再安装)如果出错,把步骤5重新运行一遍。

10.运行以下yesno

1
2
cd ~/espnet/egs/yesno/asr1
./run

出现以下结果说明安装正确