在Ubuntu上安装Caffe
时间: 2024-01-09 10:18:46 浏览: 50
在Ubuntu上安装Caffe,可以按照以下步骤进行操作:
1. 安装依赖:
- 打开终端并执行以下命令:
```
sudo apt-get update
sudo apt-get install build-essential cmake git pkg-config libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev libatlas-base-dev libopenblas-dev libopencv-dev python3-dev
```
2. 克隆Caffe源代码:
- 执行以下命令克隆Caffe的Github仓库:
```
git clone https://github.com/BVLC/caffe.git
```
3. 修改配置:
- 进入Caffe源代码目录:
```
cd caffe
```
- 复制Makefile.config.example文件并重命名为Makefile.config:
```
cp Makefile.config.example Makefile.config
```
- 使用文本编辑器打开Makefile.config文件并进行必要的修改:
```
vim Makefile.config
```
- 根据你的系统配置,可以修改以下选项:
- `CPU_ONLY`设置为`1`以仅使用CPU(如果你没有安装NVIDIA GPU驱动和CUDA)
- 如果你使用NVIDIA GPU进行加速,请确保以下选项设置正确:
- `CUDA_DIR`:CUDA安装路径
- `CUDA_ARCH`:你的NVIDIA GPU架构
- 如果你使用OpenCV 3.x,请设置以下选项:
- `OPENCV_VERSION`:3
- `INCLUDE_DIRS`和`LIBRARY_DIRS`:根据你的OpenCV 3.x安装路径进行设置
4. 编译和安装Caffe:
- 执行以下命令编译Caffe:
```
make all -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)
make pycaffe -j$(nproc)
make distribute
```
- 执行以下命令将Caffe安装到系统目录:
```
sudo make install
```
现在,你已经成功在Ubuntu上安装了Caffe。