caffe2安装

2018-06-13  本文已影响0人  哈哈哈哈_ad14

根据官网介绍,caffe2安装可以选择源码编译安装预编译安装两种方法。因为考虑到自己可能会对caffe2做改动,因此需要频繁编译安装;从而预编译安装的方式不符合这个需求,所以最后选择的是源码编译安装。
源码编译需要依赖许多第三方库,以下麻烦的地方:

这里决定使用anaconda的虚拟环境来编译安装caffe2。可以参考官网 Custom Anaconda Install

零、Anaconda虚拟环境

一、安装

1. Create a conda environment
conda create -yn caffe2-py27 && source activate caffe2-py27

2. Install required packages
conda install -y future gflags glog leveldb mkl mkl-include numpy opencv protobuf six

image.png

换成以下命令,其他包按照原来安装. 其中anaconda可以替换成其他的(inte等,看官网有什么版本)
conda install -c anaconda mkl-include

3. Clone Caffe2's source code from our Github repository
cd ~ && git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init
这一步很慢,因为文件比较大

4. Create a directory to put Caffe2's build files in
rm -rf build && mkdir build && cd build

5. Configure Caffe2's build

cmake -DCMAKE_PREFIX_PATH=~/anaconda2/envs/caffe2-py27 -DCMAKE_INSTALL_PREFIX=~/anaconda2/envs/caffe2-py27 ..

6. Compile, link, and install Caffe2
make install

上一篇下一篇

猜你喜欢

热点阅读