Install caffe in Ubuntu

Author: Zongwei Zhou | 周纵苇
Weibo: @MrGiovanni
Email: zongweiz@asu.edu


Overview

  • System: Ubuntu 16.04 LTS
  • Graphics: GeForce GTX 960M/PCIe/SSE2
  • OS type: 64-bit
  • Python: 2.7.12

Downloads

  1. CUDA Linux x86_64 Ubuntu 16.04 deb(local) 1.9GB cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb - link
  2. cuDNN log in for CUDA 8.0 cuDNN v5.1 Library for Linux cudnn-8.0-linux-x64-v5.1.tgz - link
  3. caffe github clone or download caffe-master.zip - link

Install CUDA

Open Terminator

cd Downloads/
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

then restart ubuntu.

Install cuDNN

Open Terminator

cd Downloads/cuda/
sudo su
sudo cp -P include/cudnn.h /usr/include
sudo cp -P include/cudnn.h /usr/include
sudo cp -P include/cudnn.h /usr/include

Refer to ​How can I install CuDNN on Ubuntu 16.04?

Install caffe

Open Terminator

cd Downloads/caffe-master/
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y libatlas-base-dev 
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install -y python-pip
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy
sudo apt-get install -y libopencv-dev
cp Makefile.config.example Makefile.config
sudo apt-get install kate
kate ./Makefile.config &

Uncomment

USE_CUDNN := 1
WITH_PYTHON_LAYER := 1

Change

CUDA_DIR := /user/local/cuda-8.0
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

Open Terminator

cd Downloads/caffe-master/
find . -type f -exec sed -i -e 's^"hdf5.h"^"hdf5/serial/hdf5.h"^g' -e 's^"hdf5_hl.h"^"hdf5/serial/hdf5_hl.h"^g' '{}' \;
cd /usr/lib/x86_64-linux-gnu
sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so
sudo ln -s libhdf5_serial_hl.so.10.0.2 libhdf5_hl.so 

cd Downloads/caffe-master/python
for req in $(cat requirements.txt); do pip install $req; done
cd ..

kate ./Makefile

and replace this line:

NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

with the following line

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

Open Terminator

make all
make test
make runtest
make pycaffe
make distribute

gedit ~/.bashrc

add this line at the end of the file

export PYTHONPATH=/home/zongwei/Downloads/caffe-master/python:$PYTHONPATH
export PATH=$PATH:/home/zongwei/Downloads/caffe-master/build/tools
export PATH=$PATH:/home/zongwei/git/caffe/build/lib
export PATH=".:$PATH"

Open Terminator

caffe

done!


祝好!
Weibo: @MrGiovanni</font>

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容