环境及配置: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能正常使用。