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:
...