首页 热点专区 小学知识 中学知识 出国留学 考研考公
您的当前位置:首页正文

Install caffe in Ubuntu

来源:要发发知识网

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 -
  2. cuDNN log in for CUDA 8.0 cuDNN v5.1 Library for Linux cudnn-8.0-linux-x64-v5.1.tgz -
  3. caffe github clone or download caffe-master.zip -

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

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!


显示全文