在CentOS上安裝Caffe插件,通常需要先安裝一些依賴庫和工具,然后再編譯安裝Caffe
sudo yum update -y
sudo yum groupinstall -y "Development Tools"
sudo yum install -y numpy opencv-devel python-pip protobuf-devel glog-devel gflags-devel
首先,從NVIDIA官網下載并安裝適合CentOS的CUDA Toolkit:https://developer.nvidia.com/cuda-toolkit-archive
然后,從NVIDIA官網下載并安裝適合CentOS的cuDNN庫:https://developer.nvidia.com/rdp/cudnn-archive
git clone --recursive https://github.com/BVLC/caffe.git
cd caffe
cd build
cmake -DUSE_CUDA=ON -DCMAKE_CUDA_ARCH_LIST="x86_64" -DUSE_CUDNN=ON ..
make -j$(nproc)
sudo make install
cd ..
cd python
pip install -e .
至此,您已經在CentOS上成功安裝了Caffe插件?,F在,您可以開始使用Caffe進行深度學習項目了。