在CentOS上安装Caffe主要有两种方法:通过源代码编译或通过包管理器
sudo yum update
sudo yum groupinstall "Development Tools"
sudo yum install cmake git numpy opencv-devel boost-devel leveldb-devel glog-devel
git clone https://github.com/BVLC/caffe.git
cd caffe
cd build
cmake -D CMAKE_BUILD_TYPE=Release \
-D CMAKE_INSTALL_PREFIX=/usr/local \
-D PYTHON_EXECUTABLE=/usr/bin/python2.7 \
-D WITH_CUDA=OFF \
-D WITH_CUDNN=OFF \
..
注意:这里我们禁用了CUDA和cuDNN支持,因为CentOS可能没有相应的软件包。如果需要使用GPU支持,请安装CUDA和cuDNN并按照官方文档进行配置。
make -j$(nproc)
sudo make install
cd ..
cd python
python setup.py install
至此,Caffe已成功安装在CentOS上。你可以通过以下命令验证安装是否成功:
caffe version