本篇主要介绍cuda的安装以及kaldi下的配置
cuda官网下载对应版本
https://developer.nvidia.com/compute/cuda/9.2/Prod/patches/1/cuda-repo-ubuntu1604-9-2-local-cublas-update-1_1.0-1_ppc64el
安装cuda安装包
dpkg -i cuda-repo-ubuntu1604-9-2_amd64.deb (改成对应名字)
安装cuda
apt-get update
apt-get install cuda
编译kaldi
cd 到src目录下:make clean,清除我们之前的编译结果
./configure :重新生成配置文件,这时会自动检查是否安装了cuda,这里会有显示是不是有cuda。
make
编译完成后,就可以使用GPU进行训练了。
eg:
### Testing CUDA setup with a small computation (setup = cuda-toolkit + gpu-driver + kaldi):
### Test OK!