CentOS7编译Caffe

最新推荐文章于 2021-07-04 20:57:07 发布

原创

最新推荐文章于 2021-07-04 20:57:07 发布 · 857 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#caffe

1、系统基本信息

查看操作系统的版本：$ cat /etc/redhat-release
在这里插入图片描述
查看GPU型号：$ nvidia-smi

查看CUDA的版本：cat /usr/local/cuda/version.txt

查看cuDNN的版本：nvcc -V

2、安装步骤

2.1 安装 Anaconda2

Anaconda 官网下载地址：https://www.anaconda.com/distribution/
　在这里插入图片描述下载得到：

指令安装：

$ sh Anaconda2-2019.10-Linux-x86_64.sh
之后一直Enter和输入yes即可，完成后看看光标前是否出现如下的（base）：
在这里插入图片描述
如果没有，需要source ~/.bashrc一下

2.2 创建conda虚拟环境

创建名为caffe的虚拟环境：
conda create -n caffe python=2.7
激活caffe环境：
conda activate caffe

2.3 下载 Caffe 源码

cd到$Home
git clone https://github.com/BVLC/caffe.git
访问Caffe官网：http://caffe.berkeleyvision.org/
根据官网的安装指导，安装Caffe的依赖库：
在这里插入图片描述
安装一般的依赖库：
sudo yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel
最近系统的其余依赖库：
sudo yum install gflags-devel glog-devel lmdb-devel

其余依赖项(如果没有找到)：

#glog
wget https://storage.googleapis.com/google-code-archive-downloads/v2/code.google.com/google-glog/glog-0.3.3.tar.gz
tar zxvf glog-0.3.3.tar.gz
cd glog-0.3.3
./configure
make && make install
#gflags
wget https://github.com/schuhschuh/gflags/archive/master.zip
unzip master.zip
cd gflags-master
mkdir build && cd build
export CXXFLAGS="-fPIC" && cmake .. && make VERBOSE=1
make && make install
#lmdb
git clone https://github.com/LMDB/lmdb
cd lmdb/libraries/liblmdb
make && make install

2.4 将与 CUDA 和 CUDNN 相关路径设置成环境变量

将下面两行内容添加至文件 ~/.bashrc 最后。我的cuda是9.0版本的所以：
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-9.0/
添加成功之后执行：
source ~/.bashrc

2.5 安装 BLAS（可选）

yum install atlas-devel
cd /usr/lib64/atlas
ln -sv libsatlas.so.3.10 libcblas.so
ln -sv libsatlas.so.3.10 libatlas.so

2.6 安装 Python 依赖

进入 ${CAFFE_HOME}/python/，执行
for req in $(cat requirements.txt); do pip install $req; done
注：过程中可能需要升级一些工具

2.7 更改 Caffe 的配置文件

进入 Caffe 所在目录{CAFFE_HOME}，做好备份
cp Makefile.config.example Makefile.config

打开Makefile.config文件：
vi Makefile.config
根据自己的系统配置及需求情况进行修改：

## Refer to http://caffe.berkeleyvision.org/installation.html
# Contributions simplifying and improving our build system are welcome!

# cuDNN acceleration switch (uncomment to build with cuDNN).
 USE_CUDNN := 1  ......................根据需要修改

# CPU-only switch (uncomment to build without GPU support).
# CPU_ONLY := 1

# uncomment to disable IO dependencies and corresponding data layers
# USE_OPENCV := 0
# USE_LEVELDB := 0
# USE_LMDB := 0
# This code is taken from https://github.com/sh1r0/caffe-android-lib
# USE_HDF5 := 0

# uncomment to allow M