ubuntu16.04 安裝配置matlab+python +cuda8.0+cudnn+opencv3.1的caffe環境

  網絡上有不少ubuntu上caffe配置環境的帖子,本人照着其中的許多進行了參考,都出現了或多或少的錯誤,不少地方也有差別。html

因而本身整理了下本身的安裝過程,成功進行了測試,跑通了faster-rcnn。配置環境時間爲2017.1.4 系統ubuntu16.04python

一:顯卡驅動的安裝:linux

  因爲要使用GPU,因此先要查看本身顯卡所匹配的顯卡驅動,網址:http://www.nvidia.com/Download/index.aspx?lang=en-usc++

  選擇電腦匹配的顯卡驅動,本人電腦顯卡爲GTX760,選擇以下圖所示git

 

輸入如下命令執行安裝驅動:github

sudo add-apt-repository ppa:graphics-drivers/ppa  
sudo apt-get update  
sudo apt-get install nvidia-375 //此處爲上圖的驅動版本
sudo apt-get install mesa-common-dev  
sudo apt-get install freeglut3-dev

安裝完成後,重啓電腦後輸入ubuntu

nvidia-smi

出現你的電腦gpu列表,以下所示,表示驅動安裝成功。vim

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

二:安裝配置CUDA8.0bash

  cuda官方下載地址:https://developer.nvidia.com/cuda-downloads,選擇以下:網絡

  

下載完成後,解壓到當前目錄,切換到該目錄輸入:

sh cuda_8.0.44_linux.run --override

安裝過程當中,出現不少選擇的選項,先一直按S滑到百分百,而後根據提示輸入,分別以下:

1 輸入accept接受條款
2 輸入n不安裝nvidia圖像驅動,以前已經安裝過了//此處必定要選擇n
3 輸入y安裝cuda 8.0工具
4 回車確認cuda默認安裝路徑:/usr/local/cuda-8.0
5 輸入y用sudo權限運行安裝,輸入密碼
6 輸入y或者n安裝或者不安裝指向/usr/local/cuda的符號連接
7 輸入y安裝CUDA 8.0 Samples,以便後面測試
8 回車確認CUDA 8.0 Samples默認安裝路徑:

最重要的是肯定此處不要再選擇安裝驅動。

添加環境路徑:

sudo vim ~/.bashrc  //末尾添加
export CUDA_HOME=/usr/local/cuda-8.0    
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH   
export PATH=/usr/local/cuda-8.0/bin:$PATH  

添加完成後source ~/.bashrc 使之生效。

測試cuda的例子,並獲得以下輸出:

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery  
make -j4 
sudo ./deviceQuery  

代表cuda安裝成功。

三:Cudnnv5的安裝

  從官網下載最新的cudnn,網址:https://developer.nvidia.com/rdp/cudnn-download,以下圖選擇適當的cudnn版本

  

本人下載放在Download目錄下,進入該目錄並解壓,創建軟鏈接,以下所示:

cd ./Download/
tar -xvzf cudnn-8.0-linux-x64-v5.1.tgz   
sudo cp ./cuda/include/cudnn.h /usr/local/cuda/include 
sudo cp ./cuda/lib64/lib* /usr/local/cuda/lib64

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
sudo ldconfig

四:matlab的安裝與配置

  本人使用的matlab版本位:R2014b_glnxa64_withcrack.iso

1、掛載iso(需新建matlab_iso文件夾):
   mkdir matlab_iso //主目錄
   sudo mount -o loop R2014b_glnxa64_withcrack.iso  ~/matlab_iso
2、開始安裝:
   cd ~/matlab_iso
   sudo ./install
3、選擇不聯網安裝
4、密鑰輸入,29797-39064-48306-32452
5、激活:選擇」license.lic」文件進行激活(在crack文件夾下面)
6、將libmwservices.so複製到/usr/local/MATLAB/R2014b/bin/glnxa64中:
   sudo cp libmwservices.so /usr/local/MATLAB/R2014b/bin/glnxa64/libmwservices.so

啓動matlab後出現問題,能夠參考這個帖子http://blog.csdn.net/csd_ct/article/details/51837894 。只須要將/sys/os/glnxa64/下的libstdc++.so.6 重命名位libstdc++.so.6.old.

五:Opencv3.1的安裝

  opencv3.1的下載,網址連接:http://opencv.org/downloads.html

  安裝必要項:

1 sudo apt-get update 
2 sudo apt-get install -y build-essential cmake git pkg-config 
3 sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler 
4 sudo apt-get install -y libatlas-base-dev 
5 sudo apt-get install -y --no-install-recommends libboost-all-dev 
6 sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev 
7 sudo apt-get install -y python-pip 
8 sudo apt-get install -y python-dev 
9 sudo apt-get install -y python-numpy python-scipy  
sudo apt-get install build-essential  
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev  
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev  

解壓opencv並進入該目錄:

cd opencv-3.1.0
mkdir build
cd build 
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..

期間可能會卡在ippcv的下載中,Ctrl + c中斷後,能夠到 ippicv_linux_20151201.tgz 下載。將下載的文件替換掉 OpenCV-3.1.0/3rdparty/ippicv/downloads/linux-8b449a536a2157bcad08a2b9f266828b下的同名文件。

make -j4     
sudo make install  

make過程當中,出現了graphcuts.cpp 的錯誤,進行以下修改:

sudo vim ../modules/cudalegacy/src/graphcuts.cpp 

將45行的#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER)修改成#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) || (CUDART_VERSION >= 8000)

編譯完成後注意須要運行sudo make install執行安裝

 

六:Caffe編譯

  

1 cd ~
2 git clone https://github.com/BVLC/caffe.git    //從github上git caffe

  若是沒有安裝git,能夠先使用sudo apt-get install git執行安裝。

1 cd caffe
2 sudo cp Makefile.config.example Makefile.config 
3 sudo vim Makefile.config

此時會修改caffe的編譯配置,須要修改以下地方:

 1 # USE_CUDNN := 1 修改爲: USE_CUDNN := 1 
 2 # OPENCV_VERSION := 3 修改成: OPENCV_VERSION := 3
 3 # WITH_PYTHON_LAYER := 1 修改成 WITH_PYTHON_LAYER:=1
 4 #USE_OPENCV := 0 修改爲 USE_OPENCV :=1
 5 #USE_LEVELDB := 0 修改成 USE_LEVELDB:=1
 6 #USE_LMDB := 0 修改位 USE_LMDB :=1
 7 //修改matlab目錄
 8 將MATLAB_DIR := /usr/local/MATLAB/R2014b
 9 //最終要的修改 # Whatever else you find you need goes here.下面的
10  INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 
11 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib   
12 修改成: 
  INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial 13 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial //這是由於ubuntu16.04的文件包含位置發生了變化,尤爲是須要用到的hdf5的位置,因此須要更改這一路徑

修改Makefile文件:

sudo vim Makefile
將
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)  
替換爲 
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS) 

編輯/usr/local/cuda/include/host_config.h 將119行的

#error -- unsupported GNU version! gcc versions later than 5 are not supported! 修改成 //#error -- unsupported GNU version! gcc versions later than 5 are not supported!,即添加了註釋,使之無效

進行編譯:

sudo make all -j8  
sudo make runtest -j8  

若編譯出現libcudart.so.8.0錯誤,則執行以下命令

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

此時正確狀況以下圖所示:

  

編譯matcaffe:

  執行 make matcaffe -j4

  顯示MEX completed successfully,表示編譯matcaffe成功,

  執行make mattest -j4 即運行matcaffe的測試程序,結果以下圖:

  

編譯pycaffe

1 sudo make pycaffe -j4
2 sudo make distribute -j4
3 cd python
4 for req in $(cat requirements.txt); do sudo  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple $req; done 
5 cd ../

  運行make pytest -j4進行測試,出現以下圖:

  

代表pycaffe編譯成功 在~/.bashrc 末尾添加export PYTHONPATH=/home/zf/caffe/python:$PYTHONPATH //本身對照本身路徑。

打開python導入caffe 能夠成功,以下:

 

 

七:mnist集測試:

cd ~/caffe 
./data/mnist/get_mnist.sh  
./examples/mnist/create_mnist.sh  
./examples/mnist/train_lenet.sh  

結果以下圖,正確率爲99.08%

 

八:faster-rcnn配置與測試:

  

mkdir ~/caffetest
cd caffetest
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git //下載rcnn源碼
cd py-faster-rcnn
cd lib
make 
cd ../caffe-fast-rcnn
//用以前編譯的caffe替換rcnn的Makefile.config
sudo cp ~/caffe/Makefile.config ./caffe-fastr-rcnn/
sudo make -j4 && make pycaffe
cd ../
sudo ./data/script/fetch_faster_rcnn_models.sh

此過程當中若出現cudnn錯誤,能夠作以下方式進行解決:

//在caffe-fast-rcnn目錄下 
git remote add caffe https://github.com/BVLC/caffe.git  
git fetch caffe
//可能須要github帳號 按照提示輸入git config --global user.email ~~~ 和 git config --global user.name ~~~
git stash save
git merge caffe/master 

將caffe-fast-rcnn/include/caffe/layers/python_layer.hpp的29行self_.attr("phase") = static_cast<int>(this->phase_);註釋掉

從新編譯成功後,執行demo,兩種demo爲不一樣的模型框架,

cd ~/caffetest/py-faster-rcnn
./tool/demo.py
./tool/demo.py --net zf

  若出現Check failed:error == cudaSuccess(8 vs 0) invalid device function,則將py-faster-rcnn/lib/setup.py 的第135行,將arch改成與你顯卡相匹配的數值,(好比個人GTX 760,計算能力是3.0,就將sm_35改爲了sm_30)而後刪除utils/bbox.c,nms/cpu_nms.c ,nms/gpu_nms.cpp 從新編譯便可

最終結果部分圖片以下:

  

九:總結與體會

  以前花費了時間斷斷續續的進行caffe的安裝與配置,可是總會出現大大小小的問題,可是最後才發現是第一步驅動的問題,不少博客說到的編譯器問題,在我此次安裝過程當中都沒有出現。

聽說是如今cuda8.0後對此進行了修復。最終可以徹底成功,也是很是使人高興的,也說明了,作一個事情,仍是得老老實實專心的作下去。寫這篇博客,一是爲了記念一下這一路安裝的心酸,也

是想方便之後的人,但願對他們能有幫助,少走一點彎路。加油~

  安裝過程當中用到的安裝軟件,筆者會稍後上傳到網盤,須要的能夠聯繫筆者。筆者QQ:476833883。

相關文章
相關標籤/搜索