caffe2 默认是安装到系统文件中的,如果没有权限会遇到很多问题。
本人环境是基于conda的虚拟环境,方便python包的管理,提前要装好caffe2的依赖环境。
首先是需要更改安装路径:
git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2
make && cd build
export DESTDIR="$HOME" && make -j4 install
这样把程序安装到/home/user/usr/local
下。
之后添加到系统的路径,
比如我用的是zsh,
vi .zshrc
最后加上:
export PATH="/home/yzy/usr/local/caffe2:$PATH"
export PYTHONPATH=/home/yzy/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/yzy/caffe2/build
export LD_LIBRARY_PATH=/home/yzy/usr/local/lib:$LD_LIBRARY_PATH
如果运行程序还不会成功,提示:
No module named google.protobuf.internal
这个东西来回pip都不行,最后发现使用conda安装就可以了:
conda install protobuf
哎,折腾了大半天。。