Ubuntu16.04下Nvidia+Cuda8.0+Dynet安裝教程

原文鏈接:

Ubuntu16.04下Nvidia+Cuda8.0+Dynet安裝教程 - WeiYang Blog?

godweiyang.com圖標

之前也在筆記本上裝過幾次cuda,均以失敗告終,網上的教程都沒有完全能拿來用的,多多少少都會出現一些問題。

這次終於完完全全安裝成功了,可喜可賀。。。說起來都是淚。

注意顯卡驅動安裝最新版就行了,但是cuda最好還是別安裝最新版了,裝個8.0版本吧,不然都是淚。

最終版本為ubuntu16.04 + cuda8.0 + gcc5.4。

NVIDIA驅動安裝

進入NVIDIA官網,選擇適合自己顯卡的驅動,下載後是一個.run文件。

清除之前安裝過的NVIDIA:sudo apt-get remove --purge nvidia*

禁止nouveau等驅動:sudo gedit /etc/modprob.d/blacklist.conf

加入下列語句並保存:

blacklist vga16fbblacklist nouveaublacklist rivafbblacklist nvidiafbblacklist rivatv

執行sudo update-initramfs -u並重啟reboot

ctrl+alt+f1,登錄命令行界面。

執行sudo service lightdm stop

進入NVIDIA.run目錄,運行sudo sh ./NVIDIA.run –no-x-check –no-nouveau-check –no-opengl-files

安裝過程中會報錯,直接無視。會問你要不要自動更新X配置文件,選擇是就行了。

重啟。輸入sudo nvidia-smi或者nvidia-settings,如果顯示顯卡信息,那麼恭喜你安裝成功了。

CUDA安裝

進入cuda官網,根據自己版本下載cuda.run文件。

執行sudo sh cuda.run,注意中間問你要不要安裝驅動程序,選擇no

執行sudo gedit /etc/profile

添加下列語句並保存:

export PATH=/usr/local/cuda-8.0/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH

重啟,打開/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery

執行sudo make./deviceQuery,如果出現顯卡信息,那麼cuda安裝成功了。

Dynet安裝

首先安裝Anaconda,過程就不說了,直接運行shell腳本就行了。

然後重要的地方來了,創建虛擬環境,在虛擬環境里安裝dynet!!!

執行

conda create --name python2 python=2 cython numpysource activate python2

然後CPU版本的話很簡單,直接執行pip install dynet就行了。

GPU版本執行BACKEND=cuda pip install git+https://github.com/clab/dynet#egg=dynet -i https://pypi.tuna.tsinghua.edu.cn/simple

然後就成功啦,但是運行dynet程序的時候還是會報錯,找不到libcudart庫,這時執行下面三條語句就行了:

sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfigsudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfigsudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig

然後終於可以運行了,感動哭了。。。


推薦閱讀:

ubuntu14.04安裝cuda以後卡在開機界面?
Tensorflow的GPU支持模式下的安裝要點
NVIDIA CUDA安裝包
CPU與GPU之間是如何通信的?

TAG:Ubuntu | NVIDIA英偉達 | CUDA |