在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