服务器创建虚拟环境安装caffe,caffe2安装

  • Post author:
  • Post category:其他

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

源码编译需要依赖许多第三方库,以下麻烦的地方:

root权限

第三方库容易不小心版本改变

易破坏别人的工作环境

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

创建一个新的环境,比如我的命名为 caffe2-py27

conda create -n caffe2-py27 python=2.7

后面的python=2.7是说使用2.7的python版本

启动虚拟环境: source activate caffe2-py27

退出当前环境:source deactivate

一、安装

1. Create a conda environment

conda create -yn caffe2-py27 && source activate caffe2-py27

其中-y的意思是后面的问答环节都回答yes

caffe2-py27 是我自己的环境名字,可以随意替换。

2. Install required packages

conda install -y future gflags glog leveldb mkl mkl-include numpy opencv protobuf six

在安装mkl-include的时候,会有安装不上的现象,需要指定版本

7a57e273b7f8

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

This looks for packages on your machine and figures out which functionality to include in the Caffe2 installation. The output of this command is very useful in debugging.

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

6. Compile, link, and install Caffe2

make install