kaldi-gpu配置

2018-06-25  本文已影响0人  泉泉酱

本篇主要介绍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!

上一篇下一篇

猜你喜欢

热点阅读