linux下配置caffe

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y libatlas-base-dev
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev

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

cp Makefile.config.example Makefile.config

# 启用CUDA
USE_CUDA := 1

make all -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)

make pycaffe

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

source ~/.bashrc

import caffe
import numpy as np

# 设置Caffe模式
caffe.set_mode_cpu()  # 或者 caffe.set_mode_gpu()

# 加载模型和权重
net = caffe.Net('deploy.prototxt', 'model.caffemodel', caffe.TEST)

# 准备输入数据
input_data = np.random.rand(1, 3, 224, 224)  # 示例输入

# 设置输入数据
net.blobs['data'].data[...] = input_data

# 前向传播
output = net.forward()

# 打印输出
print(output)