Caffe: 安装Caffe的最快捷方法

来源:互联网 发布:价值投资数据类app 编辑:程序博客网 时间:2024/06/02 21:14

有必要重新整理一下Caffe的安装方法,很多教程已经过时,并且装了很多没有必要的东西。本文就是关注如何能够快速地安装并使用Caffe。

1. 安装Linux

最好去买第二块硬盘,这样你会省去很多没有必要的麻烦。直接装Ubuntu 16.04,即目前最新的LTS版本,原因我会后边说。

2. 安装驱动和cuda

在系统设置的软件和更新中,安装NVIDIA和Intel的专有驱动。
这里写图片描述
装16.04的原因是,如果你有双显卡,它会自动地帮你找到你独立显卡的驱动,省得你自己瞎搞会出现各种错误。
Ubuntu下安装cuda十分简单,就一条命令:

sudo apt-get updatesudo 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 requirementsudo 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 := /usrBLAS := atlasINCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/includeLIBRARY_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

0 0
原创粉丝点击