kaldi上使用gpu以及如何安裝cuda

wbglearn發表於2014-11-17

      本部落格是在@冒頓和群裡的一個同學的指導下完成的。特此感謝……

      眾所周知,kaldi對硬體的要求,希望大家一開始都弄個好的機器,方便後面自己在kaldi裡大展身手。下面是怎麼去裝cuda和在kaldi中怎麼使用。

      特別提醒,如果顯示卡是gtx970或者980的話就不是下面這個驅動啊。大家要注意的哦。網站為:

     http://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda-repo-ubuntu1404-6-5-prod_6.5-19_amd64.deb

我網盤的地址:http://pan.baidu.com/s/1i3mpdad。也可以去這個地址下載。
      首先說明我們的顯示卡是geforce ,其他的顯示卡是否一樣不知道。以下是安裝步驟和注意事項:
     1、 安裝軟體: apt-get install ppa-purge
     2、 增加安裝源:  apt-add-repository ppa:xorg-edgers
     3、更新安裝源: apt-get update
     4、安裝nvidai 顯示卡驅動: apt-get install nvidia-current nvidia-settings
     5、下載cuda 安裝包,這裡說明下,如果是12.04就修改1404為1204哈。
          wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_6.5-14_amd64.deb
     6、安裝cuda安裝包,這個也是啊,上面下了什麼就改什麼名字啊。
           dpkg -i cuda-repo-ubuntu1404_6.5-14_amd64.deb 
     7. apt-get update
     8、安裝cuda:apt-get install cuda 
     9、在kaldi的src 中 make clean,主要是為了清楚我們之前的編譯結果
     10、./configure  :重新生成配置檔案,這時會自動檢查是否安裝了cuda,這裡會有顯示是不是有cuda。
             到src/cudamatrix 中Makefile 中修改 CUDA_ARCH =-gencode 那一行.我用的是geforce 9600 顯示卡計算能力為1.1,則修改compute=11,code=sm_11,把        TESTFILES 改為BINFILES
     11、make all 即可
        執行src/cudamatrix/目錄下執行 ./cu-vector-test,如果不報錯,則表明可以用GPU進行矩陣運算了。


     如果此方法不通,大家嘗試去這裡看下:http://blog.csdn.net/abcjennifer/article/details/23016583

後面我將更新ubuntu 14.04+cuda 6.5.謝謝……

2014.11.19晚更新:

平臺:64位 ubuntu 14.04 物理機  顯示卡是gtx 980,裝cuda 6.5.感謝@神牛的指導。

1.安裝所需要的庫:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev
libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

2.刪除所來的nvidia驅動的庫:  

sudo apt-get remove --purge nvidia*  

sudo nano /etc/modprobe.d/blacklist.conf  

在終端的末尾新增:

blacklist amd76x_edac (這個本身就有,就在這後面新增後面的哈)
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv

3.這裡首先要下載cuda_6.5.19_linux_64.run,這個我放到我百度網盤裡哈。地址:http://pan.baidu.com/s/1nt1aaed。也可以去官網下載。

關掉影像介面,按Ctrl+Alt+F1, 登陸輸入賬號和密碼,然後:

sudo service lightdm stop
chmod +x cuda_6.5.19_linux_64.run
sudo ./cuda_6.5.19_linux_64.run

這樣後就會顯示cuda的一些東西,你直接按enter到底,然後根據那個輸入acept,y等等這些啊,儘量同意那些啊。最後就告訴你:

Driver:   Installed
Toolkit:  Installed In /Usr/Local/Cuda-6.5
Samples:  Installed.

這樣基本就安裝好了啊。

 

然後再kaldi的src目錄下./configure ,會顯示cuda是不是安裝好了。然後make all,最後到src的cudamatrix目錄下執行:./cu-vector-test,就會顯示gpu什麼的啊。

至此,你就大功告成了啊。你可以執行kaldi裡的dnn模組啊。enjoy it。

下面就可以使用gpu來執行我們的kaldi程式了。相信你很期待這個速度吧,加油吧……
 歡迎使用其他顯示卡的同學分享你們的經驗,此博文將收錄到kaldi全部資料V0.5裡。

 

有任何問題可以在下面留言,我應該每天都看部落格,會回覆及時的啊。也可以去我們的kaldi群探討,歡迎您的加入。

相關文章