caffe2 安装

最新推荐文章于 2024-08-21 09:30:35 发布

zchang81

最新推荐文章于 2024-08-21 09:30:35 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： Caffe 文章标签： caffe2

本文链接：https://blog.youkuaiyun.com/zchang81/article/details/73480184

Caffe 专栏收录该内容

20 篇文章

订阅专栏

官网上的安装教程比较详细，也比较好操作Install。

依赖库

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      cmake \
      git \
      libgoogle-glog-dev \
      libprotobuf-dev \
      protobuf-compiler \
      python-dev \
      python-pip                          
sudo pip install numpy protobuf

 1
2
3
4
5
6
7
8
9
10
11
12
 1
2
3
4
5
6
7
8
9
10
11
12

GPU支持

这一部分主要是CUDA与cuDNN。

可选库

# for both Ubuntu 14.04 and 16.04
sudo apt-get install -y --no-install-recommends \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      openmpi-bin \
      openmpi-doc \
      python-pydot
sudo pip install \
      flask \
      graphviz \
      hypothesis \
      jupyter \
      matplotlib \
      pydot python-nvd3 \
      pyyaml \
      requests \
      scikit-image \
      scipy \
      setuptools \
      tornado
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

针对Ubuntu 16.04：

sudo apt-get install -y --no-install-recommends libgflags-dev

 1
2
 1
2

Clone & Build

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2
make && cd build && sudo make install
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
 1
2
3
 1
2
3

GPU测试可以执行：

python -m caffe2.python.operator_test.relu_op_test
 1
 1

这步安装完成之后，会在/usr/local/caffe2、/home/user/caffe2/build（你的build路径）路径生成caffe2的python文件，在/usr/local/lib路径生成库文件。

设置环境变量

设置正确的话，执行命令会有#后这样的输出

echo $PYTHONPATH
# export PYTHONPATH=/usr/local:$PYTHONPATH
# export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
echo $LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
 1
2
3
4
5
 1
2
3
4
5

具体设置更改对应shell的配置文件，一般来说

sudo vim /etc/profile
 1
 1

在最后加上

export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
 1
2
3
 1
2
3

就OK了。