有必要重新整理一下Caffe的安装方法,很多教程已经过时,并且装了很多没有必要的东西。本文就是关注如何能够快速地安装并使用Caffe。
1. 安装Linux
最好去买第二块硬盘,这样你会省去很多没有必要的麻烦。直接装Ubuntu 16.04,即目前最新的LTS版本,原因我会后边说。
2. 安装驱动和cuda
在系统设置的软件和更新中,安装NVIDIA和Intel的专有驱动。
装16.04的原因是,如果你有双显卡,它会自动地帮你找到你独立显卡的驱动,省得你自己瞎搞会出现各种错误。
Ubuntu下安装cuda十分简单,就一条命令:
sudo apt-get update
sudo apt-get install nvidia-cuda-toolkit
默认安装的是cuda 7.5。
3. 让cuda支持gcc 5
记住:不要降级gcc!!!虽然你要处理cuda 7.5不支持 gcc 5 以上的问题,但是不要降级gcc。因为如果你降级gcc后,其它包会出现问题,因为其它包是在gcc 5下编译的。正确方法是把cuda黑了:在/usr/include/host_config.h内把下面这几句话注释掉
// #if __GNUC__ > 5
// #error -- unsupported GNU version! gcc versions later than 5 are not supported!
// #endif /* __GNUC__ > 5 */
4.安装Caffe
Caffe的安装方法也比较简单
1. 安装Caffe所需要依赖包
sudo apt-get install build-essential # basic requirement
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 #required by caffe
2. 安装Atlas
sudo apt-get install libatlas-base-dev
3. 安装opencv
sudo apt-get install libopencv-dev
4. 下载Caffe
git clone https://github.com/BVLC/caffe.git
5. 修改Makefile
--- NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
+++ NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
上面的意思是把第一行替换成第二行
6. 修改Makefile.config
cp Makefile.config.example Makefile.config
这里仅标出需要修改的地方。
CUDA_DIR := /usr
BLAS := atlas
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
参考:
1. Ubuntu 14.04下的安装教程
2. 双显卡笔记本安装Cuda
3. 主要看它黑Cuda过程
4. 看它如何修改Makefile文件
5. 如何在Ubuntu上安装Opencv