在Ubuntu 18.04安装PyTorch-kaldi
时间: 2024-04-06 16:13:21 浏览: 97
在计算机上安装和配置 PyTorch
安装 PyTorch-kaldi 需要以下步骤:
1. 安装依赖项:
```
sudo apt-get update
sudo apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
libatlas-base-dev \
libboost-all-dev \
libgflags-dev \
libgoogle-glog-dev \
libhdf5-serial-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libprotobuf-dev \
libsnappy-dev \
protobuf-compiler \
python-dev \
python-numpy \
python-pip \
python-setuptools \
python-scipy \
wget
```
2. 安装 Kaldi:
```
git clone https://github.com/kaldi-asr/kaldi.git kaldi --origin golden
cd kaldi/tools
make -j 4
cd ../src
./configure --shared --use-cuda=no
make depend -j 4
make -j 4
```
3. 安装 PyTorch:
```
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu110/torch_stable.html
```
4. 下载并安装 PyTorch-kaldi:
```
git clone https://github.com/mravanelli/pytorch-kaldi.git
cd pytorch-kaldi/src
```
修改 Makefile.config 文件,将以下两行的注释符号 # 去掉:
```
KALDI_ROOT?=../../kaldi
CUDA_ROOT=/usr/local/cuda
```
然后运行以下命令:
```
make depend
make
```
安装完成后,您可以尝试运行示例文件以确保一切正常:
```
cd ../examples/simple_vad
python main.py
```
如果一切正常,您应该能够看到输出结果。
阅读全文