Caffe: 安装Caffe的最快捷方法

最新推荐文章于 2024-05-03 15:15:30 发布

斡艾汀余

最新推荐文章于 2024-05-03 15:15:30 发布

阅读量4.5k

点赞数 1

分类专栏： Caffe 文章标签： cuda Caffe

本文链接：https://blog.csdn.net/lry_xueshu/article/details/53403700

版权

Caffe 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

有必要重新整理一下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