網絡上有不少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。