caffe学习:在ubuntu下安装caffe

环境及配置:ubuntu14.04 + cuda8.0 + cudnn5,NVIDIA gtx1080

一,安装NVIDIA驱动

打开系统设置>软件和更新>附加驱动,选择NVIDIA的驱动,版本稍微高一点就好,这里选择的384版本的驱动,然后点应用更改,等待自动安装完成。
这里写图片描述

二,安装cuda8.0和cudnn5

1.安装cuda

在NVIDIA官网下载需要注册账号,这里就直接上网盘链接了:https://pan.baidu.com/s/1Tj5oZ5BM_z9AUzo5cHDEIQ 提取密码:tq4q
打开terminal,切换到下载cuda的目录下输入:

sudo sh cuda_8.0.61_375.26_linux.run

一直按空格键跳过说明,然后中间选择不安装375版本的驱动即可(已经装过384驱动了)

Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n

剩下的一直选择yes即可

2.安装cudnn5

还是上网盘链接吧:https://pan.baidu.com/s/14zBa75n4njxRnzKr3TCWcQ 提取密码:sird
解压后打开terminal输入:

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
sudo cp cudnn5/include/cudnn.h /usr/local/cuda/include/
sudo cp cudnn5/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 /usr/local/cuda/lib64

三,安装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

1.从github上下载caffe:https://github.com/BVLC/caffe 并解压
2.进入caffe目录下,复制一份Makefile.config.example,修改名字为Makefile.config
在Makefile.config中更改

USE_CUDNN := 1
USE_LEVELDB := 1
USE_LMDB := 1
WITH_PYTHON_LAYER := 1
MATLAB_DIR := /home/user/matlab #添加自己的matlab路径
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

3.在Makefile中更改

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

替换为:

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

执行:

make all -j8

等待编译完成。
4.编译matcaffe和pycaffe

make matcaffe
make pycaffe

五,将caffe添加到环境变量.

sudo gedit ~/.bashrc

将下面代码添加到最后一行(路径修改为自己的caffe路径),保存并退出

export PYTHONPATH=/home/user/caffe/python:$PYTHONPATH

执行:

source ~/.bashrc

使环境变量生效。
进入python:

import caffe

未报错说明caffe能正常使用。

Guess you like

Origin blog.csdn.net/qi_sama/article/details/82378252