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 是否安裝成功。