因为在firefly3399上要进行人脸识别,而模型使用的是caffe2,所以这里记录一下安装的过程.
一.安装
1.更新源,安装相关依赖
sudo apt-get update sudo apt-get install -y --no-install-recommends \ build-essential \ cmake \ git \ libgoogle-glog-dev \ libgtest-dev \ libiomp-dev \ libleveldb-dev \ liblmdb-dev \ libopencv-dev \ libopenmpi-dev \ libsnappy-dev \ libprotobuf-dev \ openmpi-bin \ openmpi-doc \ protobuf-compiler \ python-dev \ python-pip sudo pip install \ future \ numpy \ protobuf
2.安装共享动态库
sudo apt-get install -y --no-install-recommends libgflags-dev
3.下载caffe2的源码,因为现在caffe被并在了pytorch里,所以我们git的是pytorch的资源
git clone --recursive https://github.com/pytorch/pytorch.git
4.进入pytorch目录
cd pytorch
git submodule update
--init
5.进行cmake,cmake 的时候最好加上sudo,不然容易报错
sudo cmake ..
6.编译链接安装caffe2
sudo make install
二.测试
现在caffe2就已经安装好了,现在来测试caffe2是否安装成功
扫描二维码关注公众号,回复:
1925178 查看本文章
cd ~
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
如果是"Failure",可以去这个链接查看原因:https://caffe2.ai/docs/faq.html#why-do-i-get-import-errors-in-python-when-i-try-to-use-caffe2如果返回"Success",那么说明安装已经成功了.
博主当时安装了但是返回是Failure,查看python2.7的dist-packages发现什么都没有,但是我确实安装了的呀.查找原因才知道原来是因为之前安装了Tensorflow,而firefly3399安装Tensorflow的时候会把pip改为pip3,所以上面pip其实是pip3在进行安装,如果想在python2中进行安装,那么把上面的pip改为pip2就可以了.