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
若系统为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
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
sudo pip install \
flask \
graphviz \
hypothesis \
jupyter \
matplotlib \
pydot python-nvd3 \
pyyaml \
requests \
scikit-image \
scipy \
setuptools \
tornado
首先从网上下载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
只要没有报错我们就大功告成了。
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
这个我们在进行编译之前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
整个安装过程中没有出现问题,但是在验证的时候就会出现错误,环境变量也是安装教程所设置的,始终没有找到原因,重新安装折腾好几次。最后才发现,在终端中输入命令使其输出环境变量
echo $PYTHONPATH
经过查看才知道,Ubuntu中环境变量法并不是安装文档的前后顺序来进行排列的,是根据你修改的顺序,后来修改的环境变量会在前边。
出现种种问题时,先将其他环境变量注释即可。
sudo pip install \
future \
numpy \
protobuf
安装时,出现了这个错误,按照提示运行下边的命令
pip install --upgrade pip
这样还不可以还需要安装setuptools
sudo pip install setuptools
参考:
caffe2 安装教程