centos安装caffe路径设置

在 CentOS 上安装 Caffe 时，需要正确设置路径以确保系统能够找到所需的库和可执行文件

1. 首先，确保已安装必要的依赖项。在终端中运行以下命令：

sudo yum install -y numpy python-pip matplotlib

2. 安装 CUDA 和 cuDNN。根据您的 GPU 型号和驱动程序版本，从 NVIDIA 官方网站下载并安装适当的 CUDA 和 cuDNN 工具包。安装完成后，记下 CUDA 和 cuDNN 的安装路径，例如 /usr/local/cuda 和 /usr/local/cuda-10.2/lib64。

3. 下载 Caffe 源代码。访问 Caffe 的 GitHub 仓库（https://github.com/BVLC/caffe），然后点击 “Code” 按钮，选择 “Download ZIP”。解压下载的 ZIP 文件到一个目录，例如 /opt/caffe。

4. 在 Caffe 源码目录中创建一个名为 build 的文件夹，然后进入该文件夹：

cd /opt/caffe
mkdir build
cd build

5. 配置 Caffe 编译选项。运行以下命令，将 <CUDA_PATH> 替换为实际的 CUDA 安装路径，将 <CUDNN_PATH> 替换为实际的 cuDNN 安装路径：

cmake -DCMAKE_INSTALL_PREFIX=/usr/local .. -DCMAKE_CUDA_ARCH_LIST="x86_64" -DCMAKE_CUDA_FLAGS="-I<CUDA_PATH>/include -L<CUDA_PATH>/lib64" -DCMAKE_CUDNN_FLAGS="-I<CUDNN_PATH>/include" -DCMAKE_CXX_FLAGS="-std=c++11"

6. 编译并安装 Caffe：

make -j$(nproc)
sudo make install

7. 设置环境变量。编辑 ~/.bashrc 或 ~/.bash_profile 文件，添加以下内容：

export PATH=/usr/local/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
export Caffe_ROOT=/usr/local/caffe

保存文件后，运行以下命令使更改生效：

source ~/.bashrc

现在，您已经在 CentOS 上成功安装了 Caffe，并设置了正确的路径。可以使用 caffe 命令检查安装是否成功。