centos安装caffe依赖库呢

在 CentOS 上安装 Caffe 及其依赖库，可以按照以下步骤进行：

1. 更新系统： 首先，确保你的系统是最新的。

   sudo yum update -y
   

2. 安装必要的开发工具和编译器： 安装一些必要的开发工具和编译器。

   sudo yum groupinstall "Development Tools" -y
   sudo yum install cmake git -y
   

3. 安装 CUDA 和 cuDNN： Caffe 需要 NVIDIA 的 CUDA 和 cuDNN 库来加速计算。你需要从 NVIDIA 官方网站下载并安装适合你 GPU 版本的 CUDA 和 cuDNN。

   • 下载 CUDA Toolkit：

     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
     sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
     wget http://developer.download.nvidia.com/compute/cuda/11.4.2/Prod/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-450.51.06-1_amd64.deb
     sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-450.51.06-1_amd64.deb
     sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
     sudo apt-get update
     sudo apt-get -y install cuda
     

   • 下载 cuDNN：

     wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5.32/Production/11.4_20211031/cudnn-11.4-linux-x64-v8.0.5.32.tgz
     tar -xvf cudnn-11.4-linux-x64-v8.0.5.32.tgz
     sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
     sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
     sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
     

4. 安装 Python 和 pip： 确保你已经安装了 Python 和 pip。

   sudo yum install python3 -y
   sudo yum install python3-pip -y
   

5. 安装其他依赖库： 安装 Caffe 所需的其他依赖库。

   sudo pip3 install numpy matplotlib scipy
   

6. 克隆 Caffe 仓库： 克隆 Caffe 仓库到本地目录。

   git clone --recursive https://github.com/BVLC/caffe.git
   cd caffe
   

7. 编译和安装 Caffe： 进入 Caffe 目录并运行 CMake 配置和编译命令。

   cd ..
   mkdir build && cd build
   cmake -D CMAKE_BUILD_TYPE=Release \
         -D CMAKE_INSTALL_PREFIX=/usr/local \
         -D CUDA_ARCH_LIST="6.1" \
         -D WITH_CUDA=ON \
         -D WITH_CUDNN=ON \
         -D WITH_OPENCL=OFF \
         -D WITH_REDIS=OFF \
         -D WITH_ZMQ=OFF \
         -D WITH_MPI=OFF \
         ..
   make -j$(nproc)
   sudo make install
   

8. 设置环境变量： 设置一些环境变量以便系统能够找到 Caffe 和 CUDA。

   echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
   echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
   source ~/.bashrc
   

完成以上步骤后，你应该能够在 CentOS 上成功安装 Caffe 及其依赖库。你可以通过编写一个简单的 Python 脚本来测试 Caffe 是否安装成功。