ubuntu20 安装 caffe-cuda

首先，在 Ubuntu 20 上安装 Caffe-CUDA 需要满足一些先决条件。请确保你已经安装了 CUDA 和 cuDNN，并且已经正确配置了它们。然后，按照以下步骤进行安装：

1. 安装依赖项：

   sudo apt-get update
   sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev

2. 克隆 Caffe 仓库：

   git clone https://github.com/BVLC/caffe.git

3. 进入 Caffe 目录：

   cd caffe

4. 编辑 Makefile.config 文件：

   cp Makefile.config.example Makefile.config
   nano Makefile.config

确保按照你的配置修改以下行：
- `USE_CUDNN := 1` （如果已安装 cuDNN）
- `CUDA_DIR := /usr/local/cuda` （CUDA 的安装路径）
- `INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial` （根据你的系统配置修改）
- `LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial` （根据你的系统配置修改）

5. 编译 Caffe：

   make all -j$(nproc)  # 使用所有可用的 CPU 内核
   make test -j$(nproc)
   make runtest -j$(nproc)

6. 安装 Caffe：

   sudo make install

7. 配置环境变量：

   echo 'export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH' >> ~/.bashrc
   source ~/.bashrc

现在，你应该已经成功在 Ubuntu 20 上安装了 Caffe-CUDA。你可以尝试运行一些示例代码来验证安装是否成功。记得替换 `/path/to/caffe` 为你的 Caffe 目录路径。希望对你有所帮助！