涉及到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镜像

结果Non-local I3D 配置记录

相关文章:

  • 2021-08-25
  • 2021-07-16
  • 2021-07-31
  • 2022-01-14
  • 2021-12-10
  • 2021-10-08
  • 2021-06-02
猜你喜欢
  • 2021-12-14
  • 2021-11-15
  • 2021-08-11
  • 2022-01-17
  • 2021-05-30
  • 2021-11-18
相关资源
相似解决方案