Caffe2安装_caffe教程

Caffe2安装_caffe教程一、依赖库的安装首先1.首先需要安装Ubuntu16.04与14.04都需要的依赖库。sudoaptupdatesudoaptinstall-y–no-install-recommends\build-essential\cmake\git\libgoogle-glog-dev\lib…

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全家桶1年46,售后保障稳定

一、依赖库的安装首先

1.首先需要安装Ubuntu 16.04与14.04都需要的依赖库。

sudo apt update
sudo apt 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

Jetbrains全家桶1年46,售后保障稳定

2.分版本安装依赖库

若系统为Ubuntu 16.04时,需要安装libgflags-dev

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

若系统为Ubuntu 14.04时,需要安装libgflags2

sudo apt-get install -y --no-install-recommends libgflags2

3.安装一些可选的相关性

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 -H pip install --upgrade pip 

4.更新一下pip

sudo pip install \       flask \       graphviz \       hypothesis \       jupyter \       matplotlib \       pydot python-nvd3 \       pyyaml \       requests \       scikit-image \       scipy \       setuptools \       tornado

二、Caffe2安装

首先从网上下载Caffe2安装库

git clone --recursive https://github.com/caffe2/caffe2.git

下载完以后进行安装

cd caffe2
mkdir build
make -j
cd build
sudo make install

安装结束以后我们可以使用下面的程序检测安装十分偶成功

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

接下来我们检测GPU是否安装成功,如果没有安装GPU,会显示warning说是CPU only,最后test pass。这样说明安装成功

python -m caffe2.python.operator_test.relu_op_test

环境变量设置

首先进入环境变量

sudo gedit ~/.bashrc

在文档的最后添加

export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/lemon(user)/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

添加完以后更新一下环境变量

source ~/.bashrc

到此我们安装就已经结束了,可以去python中

import caffe2

只要没有报错我们就大功告成了。

安装过程中出现到问题

1.在caffe2文件夹直接进行编译时,出现内存不足问题

c++: internal compiler error: Killed (program cc1plus)
Please submit a full bug report,
with preprocessed source if appropriate.
See <file:///usr/share/doc/gcc-5/README.Bugs> for instructions.
caffe2/CMakeFiles/caffe2.dir/build.make:6134: recipe for target 'caffe2/CMakeFiles/caffe2.dir/queue/rebatching_queue_ops.cc.o' failed
make[3]: *** [caffe2/CMakeFiles/caffe2.dir/queue/rebatching_queue_ops.cc.o] Error 4
make[3]: *** Waiting for unfinished jobs....

这里写图片描述

在网上查询以后发现是内存不足,我使用的是E5 2695 V3,内存32G,28线程全开时内存不足,把线程缩小到24线程时可以编译成功。

make -j24

2.在caffe2进行编译的时候没有找到 CUDA 和 cuDNN 路径

这个我们在进行编译之前cmake一下,提前设置这两个的路径,在caffe2目录下

cmake \
  -DCUDA_TOOLKIT_ROOT_DIR=/path/to/cuda/toolkit/dir \   -DCUDNN_ROOT_DIR=/path/to/cudnn/root/dir

例如我的设置为

cmake \
  -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-8.0 \   -DCUDNN_ROOT_DIR=/usr/local/cuda

3.python -c ‘from caffe2.python import core’ 2>/dev/null && echo “Success” || echo “Failure”出现Failure

整个安装过程中没有出现问题,但是在验证的时候就会出现错误,环境变量也是安装教程所设置的,始终没有找到原因,重新安装折腾好几次。最后才发现,在终端中输入命令使其输出环境变量

echo $PYTHONPATH

经过查看才知道,Ubuntu中环境变量法并不是安装文档的前后顺序来进行排列的,是根据你修改的顺序,后来修改的环境变量会在前边。
出现种种问题时,先将其他环境变量注释即可。

4.Command “python setup.py egg_info” failed with error code 1 in /tmp/pip-build-lak8RD/future/

这里写图片描述
在进行

sudo pip install \       future \       numpy \       protobuf

安装时,出现了这个错误,按照提示运行下边的命令

pip install --upgrade pip

这里写图片描述

这样还不可以还需要安装setuptools

sudo pip install setuptools

这里写图片描述
这样再接着安装命令就可以顺利通过了。
这里写图片描述

参考:

caffe2 安装教程

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/213584.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)
blank

相关推荐

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号