歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux綜合 >> Ubuntu Linux >> 在ubuntu14.04上配置cuda_caffe_cudnn_anaconda_digits,caffecudacudnn

在ubuntu14.04上配置cuda_caffe_cudnn_anaconda_digits,caffecudacudnn

日期:2017/3/3 18:08:54   编辑:Ubuntu Linux

在ubuntu14.04上配置cuda_caffe_cudnn_anaconda_digits,caffecudacudnn

在ubuntu14.04上配置cuda_caffe_cudnn_anaconda_digits,caffecudacudnn


參考網上的很多網站,以這篇為主:http://blog.csdn.net/yhl_leo/article/details/50961542

這篇算是自己對caffe學習的一個總結系列的開頭。首先因為caffe的依賴項比較多,配置起來也比較麻煩。這篇算是比較詳盡地把caffe的各種相關配置說清楚。轉載請注明出處。

推薦ubuntu14.04版本,因為digits的支持比較好。顯卡支不支持GPU加速要搞清楚啊,不支持就全用CPU算吧,別浪費時間折騰cuda了。

大家基本要按照官方教程上面來。但是官方教程有的坑沒有點出來的,本文也一並列上了。

一、顯卡的安裝

禁用nouveau驅動,nouveau是ubuntu自帶的對nivida的開源驅動,對安裝nvidia的官方驅動會有問題,所以先將其禁用。

按Ctrl+Alt+F1 進入tty1控制台,輸入

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

在裡面寫上

blacklist nouveau

options nouveau modeset=0

按esc 輸入:wq 保存退出

ps:vim編輯器很好用啊,還不會的小伙伴要抓緊時間學啊。

執行

lspci | grep nouveau

查詢pci總線中是否還有nouveau,grep後面接正則表達式過濾。

什麼都沒有說明禁用成功。

重啟後登錄時,可能會循環出現填寫登錄密碼,不能進入系統的情況,按Ctrl+Alt+F1,登錄,

卸載顯卡方法:

sudo apt-get remove --purge nvidia*

安裝顯卡驅動:

$ sudo add-apt-repository ppa:xorg-edgers/ppa

$ sudo apt-get update

$ sudo apt-get install nvidia-352

執行sudo start lightdm 輸入密碼能看到桌面就ok啦

ps:顯卡驅動掛掉之後很可能你就看不到系統界面了,啟動之後顯示器一團漆黑,其實系統是在運行的。我的做法是先提前裝好了SSH服務,碰到這種情況就通過另一台電腦SSH登陸進去安裝驅動。

二、CUDA安裝

CUDA官網下載。其實應該先下載好的,如果沒來得及用圖形界面下載,用wget, curl什麼的下載也一樣。 據說331的驅動有坑啊,大家不要用那個驅動。下面安裝一些依賴項/

sudo service lightdm stop

sudo apt-get install g++

sudo apt-get install git

sudo apt-get install freeglut3-dev

首先在官網上下載安裝文件(鏈接前文已經提供):

我下載的是deb本地安裝文件,下載完成後,按照文檔提示的命令安裝:

sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

sudo apt-get update

sudo apt-get install cuda

下載了下方的Installation Guide for Linux,裡面有關於環境變量的設置方法:

如果有,則說明安裝成功。沒有可以按照下面方法卸載:

sudo /usr/local/cuda-7.5/bin/uninstall_cuda_7.5.plsudo /usr/bin/nvidia-uninstall

最後,配置環境變量,我們直接放在系統配置文件profile裡面,先打開profile文件

sudo vi /etc/profile

在最後面加入兩行代碼:

export PATH=/usr/local/cuda-7.5/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

執行 source /etc/profile 使環境變量生效

至此cuda安裝完畢

三、CUDNN安裝

CUDNN是給CUDA加速的。cuDNN是GPU加速計算深層神經網絡的庫。cudnn官網下載。貌似下載要先注冊,審核通過還要幾天時間。下載好之後解壓。

執行如下命令:

sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.tgz && cd cuda/include && sudo cp *.h /usr/local/include/ && cd ../lib64 && sudo cp lib* /usr/local/lib/ && cd /usr/local/lib && sudo chmod +r libcudnn.so.4.0.7 && sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4 && sudo ln -sf libcudnn.so.4 libcudnn.so && sudo ldconfig

這樣CUDNN就安裝完畢了,是不是很簡單啊。

四、Anaconda的安裝

到https://www.continuum.io/downloads 下載anaconda,推薦使用linux版的python 2.7版本,因為tensorflow中的有些東西不支持python3.5(如cPickle)。

下載成功後,在終端執行(2.7版本):

# bash Anaconda2-4.1.1-Linux-x86_64.sh

或者3.5 版本:

# bash Anaconda3-4.1.1-Linux-x86_64.sh

在安裝的過程中,會問你安裝路徑,直接回車默認就可以了。有個地方問你是否將anaconda安裝路徑加入到環境變量(.bashrc)中,這個一定要輸入yes

安裝成功後,會有當前用戶根目錄下生成一個anaconda2的文件夾,裡面就是安裝好的內容。在終端可以輸入

conda info 來查詢安裝信息

輸入conda list 可以查詢你現在安裝了哪些庫,常用的python, numpy, scipy名列其中。如果你還有什麼包沒有安裝上,可以運行

conda install *** 來進行安裝(***代表包名稱),如果某個包版本不是最新的,運行 conda update *** 就可以了。

五、caffe的安裝

首先先安裝opencv,推薦2.4的版本。opencv1.x是純C語言編寫的,2.x c和c++的包都有,opencv3是只用c++寫的。為了別人寫的代碼也能正常運行,還是推薦裝2.x。

opencv2.4安裝很簡單了,下載下來解壓,然後進入目錄make, sudo make install就搞定了。

caffe官方下載 基本按照官方安裝指南就可以了,l另外一個方法就是命令行下載

下載caffe:

sudo git clone https://github.com/BVLC/caffe.git

如果你沒安裝Git,請閱讀博客:Ubuntu Git安裝與使用。

然後編譯caffe:先安裝依賴:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

$ sudo apt-get install --no-install-recommends libboost-all-dev

$ sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

$ sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

賈揚清大神還說了,大家一定要看清楚啊!

下面開始配置caffe並且開始編譯了。

sudo cp Makefile.config.example Makefile.config

# Adjust Makefile.config (for example, if using Anaconda Python, or if cuDNN is desired) make all make test make runtest

配置文件有幾點要注意:

# cuDNN acceleration switch (uncomment to build with cuDNN).

USE_CUDNN := 1

# CPU-only switch (uncomment to build without GPU support).

# CPU_ONLY := 1BLAS choice:# atlas for ATLAS (default)

# mkl for MKL

# open for OpenBlasBLAS := atlas

# Custom (MKL/ATLAS/OpenBLAS) include and lib directories.

# Leave commented to accept the defaults for your choice of BLAS

# (which should work)!

# BLAS_INCLUDE := /path/to/your/blas

# BLAS_LIB := /path/to/your/blas

這幾個地方根據自己情況選擇。

make all沒有問題的話,caffe就算成功了啊。

保存後重新編譯: (如果電腦有8個線程的話,就在後面加 -j8 ,可以加快編譯速度)

sudo make clean

sudo make all -j8

sudo make test -j8

sudo make runtest -j8

本人編譯的時候最後一步的時候,出現這樣的錯誤:

libcudart.so.7.5 cannot open shared object file: No such file or directory

網上有一堆介紹,怎麼設置環境變量的(我們分明已經設置過了),可以這樣檢查是否已經添加環境變量:

$ echo $PATH$ echo $LD_LIBRARY_PATH

可以看到,環境變量已經添加好。

解決方法是這樣,將一些文件復制到/usr/local/lib文件夾下:

sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig

sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig

sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig

再次,嘗試sudo make runtest命令,出現如下:

至此,caffe安裝完成。

將caffe路徑導入環境變量,執行命令如下:

sudo vi ~/.bashrc

在最後加上 export PYTHONPATH=/home/***/caffe/python:$PYTHONPATH

export CAFFE_HOME=/home/***/caffe:$CAFFE_HOME

之後執行 sudo ldconfig 來生效

接著在caffe裡面執行sudo make pycaffe ,沒有錯誤就OK了,測試caffe是否成功,

在終端輸入 python 回車,import caffe 沒有錯誤表示ok,

如果出現No module named google.protobuf.internal

解決辦法參考鏈接:http://www.th7.cn/system/lin/201605/164288.shtml

sudo chmod 777 -R anaconda2(文件夾) 改變權限 然後
conda install protobuf

就搞定啦~

DIGITS安裝

參考鏈接:http://www.cnblogs.com/denny402/p/5136155.html

一、安裝digits 3.0

digits是運行在cuda和caffe基礎上的,所以要先配置好cuda+caffe那是毫無疑問的了。還不會配置的,請參考:Caffe學習系列(1):安裝配置ubuntu14.04+cuda7.5+caffe+cudnn

打開一個終端,依次運行下列命令:

cd

sudo -s

進入當前用戶根目錄,並切換到超級用戶(符號由$變成#,不用每句都輸sudo)

CUDA_REPO_PKG=cuda-repo-ubuntu1404_7.5-18_amd64.deb &&

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/$CUDA_REPO_PKG &&

sudo dpkg -i $CUDA_REPO_PKG

接著

ML_REPO_PKG=nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb &&

http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/$ML_REPO_PKG &&

sudo dpkg -i $ML_REPO_PKG

apt-get update

apt-get install digits

http://xxxxxx/Linuxjc/1170257.html TechArticle

Copyright © Linux教程網 All Rights Reserved