涉及到caffe/caffe2的配环境都挺麻烦的,尤其是cuda8+python2.7的,选择使用docker进行环境配置
安装步骤
1.pull docker镜像 docker pull bvlc/caffe:gpu,创建容器
2.完整caffe2仓库
3.流程
git clone --recursive https://github.com/CivilNet/video_nonlocal_net_caffe2
cd video_nonlocal_net_caffe2
mkdir build
cd build
cmake ..
make -j16
cd ..
python setup_caffe2.py install
export PYTHONPATH=[PATH TO Non-local lib]
Tip:如果需要opencv或者提示缺什么库,安装即可
pip install opencv-python
pip速度过慢请使用tsinghua镜像