Ubuntu caffe搭建
1、安装NVIDIA驱动和CUDA
2、安装caffe依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler python-pip python-dev python-opencv
3、安装Atlas
sudo apt-get install libatlas-base-dev
4、安装cudnn
https://developer.nvidia.com/cudnn
解压
tar -xzvf cudnn-**
cd cuda/lib64
sudo cp * /usr/local/cuda/lib64
cd ../include
sudo cp cudnn.h /usr/local/cuda/include
更新软链接
/usr/local/cuda/lib64
5、安装caffe
a)进入caffe根目录, 首先复制一份 Makefile.config , 然后修改里面的内容,主要需要修改的参数包括
USE_CUDNN 是是否使用cudnn加速
CPU_ONLY 是否只使用CPU模式,没有GPU没安装CUDA的同学可以打开这个选项
BLAS (使用intel mkl还是atlas)
MATLAB_DIR 如果需要使用MATLAB wrapper的同学需要指定matlab的安装路径, 如我的路径为 /usr/local/MATLAB/R2013b (注意该目录下需要包含bin文件夹,bin文件夹里应该包含mex二进制程序)
DEBUG 是否使用debug模式,打开此选项则可以在eclipse或者NSight中debug程序
b)编译
make –j8 all
make -j8 test
make runtest
注意 -j8 是指使用几个线程来同时编译, 可以加快速度, j后面的数字可以根据CPU core的个数来决定。
c)CMake 生成cmake程序的所有引用
在CAFFE_ROOT目录下
mkdir cmakebuild
cd cmakebuild
cmake ..
make all -j8
make runtest
d)以下步骤可选择:
编译Matlab wrapper
执行如下命令
make matcaffe
然后就可以跑官方的matlab demo啦。
编译Python wrapper
make pycaffe