美文网首页
caffe2安装

caffe2安装

作者: 哈哈哈哈_ad14 | 来源:发表于2018-06-13 09:12 被阅读0次

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

    • root权限
    • 第三方库容易不小心版本改变
    • 易破坏别人的工作环境

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

    零、Anaconda虚拟环境

    • 创建一个新的环境,比如我的命名为 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的时候,会有安装不上的现象,需要指定版本
    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

    相关文章

      网友评论

          本文标题:caffe2安装

          本文链接:https://www.haomeiwen.com/subject/yqhorftx.html