CentOS7安装Caffe的教程详解

  • Post category:Linux

CentOS7安装Caffe的教程详解

1. 安装依赖环境

Caffe在编译时需要一些依赖库,包括openblas、CUDA和cuDNN等。使用以下命令来安装openblas:

sudo yum install -y openblas-devel

使用以下命令安装CUDA和cuDNN:

sudo yum install -y nvidia-driver-latest-dkms cuda-dnn-devel-8-0

2. 下载和编译Caffe

下载最新版的Caffe代码:

git clone https://github.com/BVLC/caffe.git

进入到Caffe的根目录,使用以下命令进行编译:

cd caffe
cp Makefile.config.example Makefile.config
# 编辑Makefile.config文件,修改相关配置,例如是否使用CUDA加速等
make -j8 all
make -j8 test
make runtest

编译过程可能需要一段时间,取决于您的机器配置。

3. 配置环境变量

为了让您方便地使用Caffe,您需要将编译后的可执行文件添加到系统的PATH环境变量中,以便全局使用。在.bashrc文件中添加以下几行:

export PATH=/path/to/caffe/build/tools:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

示例说明:

示例一:在Caffe中使用GPU加速

编辑Makefile.config文件,将以下两行的注释符号去掉:

# USE_CUDA := 1
# USE_CUDNN := 1

执行make命令,Caffe会自动使用CUDA和cuDNN进行加速。

示例二:测试Caffe的安装是否成功

使用以下命令来测试Caffe的安装是否成功:

./build/tools/caffe time -model models/bvlc_alexnet/deploy.prototxt -gpu 0

如果输出类似于以下内容,则说明Caffe安装成功:

I1012 01:24:53.862786 18844 caffe.cpp:244] Starting Optimization
I1012 01:24:53.862800 18844 solver.cpp:50] Initializing solver from parameters:
...