【问题标题】:Cannot import caffe2 from caffe2.python?无法从 caffe2.python 导入 caffe2?
【发布时间】:2018-01-01 14:24:04
【问题描述】:

我正在尝试在 macbook pro 上安装 caffe2。我按照官方说明: https://caffe2.ai/docs/getting-started.html?platform=mac&configuration=compile#python-configuration

我成功地从源代码构建它,但是当我尝试命令时:

~/anaconda2/bin/python -c 'from caffe2.python import core' 2>/dev/null & echo "Success" | echo "Failure"

上面写着:

Failure
'~/anaconda2/bin/python -c 'from…' has ended

我不知道它有什么问题。 我的电脑有两个不同版本的python。默认的是'/usr/bin/python'目录中的python 2.7.10。另一个是'~/anaconda2/bin'目录中的python 2.7.13。我不知道它用的是哪一种。使用命令:

mkdir build && cd build
cmake -DUSE_CUDA=OFF ..
sudo make install

我安装成功了。我还在~/.bash_profile中设置了PYTHONPATH环境变量export PYTHONPATH="~/anaconda2/bin/"并执行命令:

sudo install_name_tool -change libpython2.7.dylib ~/anaconda2/lib/libpython2.7.dylib /usr/local/caffe2/python/caffe2_pybind11_state.so

到目前为止一切正常。但是当我执行时:

~/anaconda2/bin/python -c 'from caffe2.python import core' 2>/dev/null & echo "Success" | echo "Failure"

显示“失败”,我也试过了:

python -c 'from caffe2.python import core' 2>/dev/null & echo "Success" | echo "Failure"

它没有工作。我该怎么办?

【问题讨论】:

    标签: python python-2.7 anaconda caffe


    【解决方案1】:

    conda/build.sh 中,将BLAS=MKL 替换为BLAS=Eigen

    CMAKE_ARGS+=("-DBLAS=MKL") # gives unresolved dnnBatchNormalizationCreateForward_v2_F32
    CMAKE_ARGS+=("-DMKL_INCLUDE_DIR=$CONDA_PREFIX/include")
    CMAKE_ARGS+=("-DBLAS=Eigen")
    

    conda/meta.yaml 中注释掉对 MKL 构建的引用:

    # - mkl-include
    # - mkl
    

    运行:

    # - mkl-include
    # - mkl
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2018-09-28
      • 2021-02-15
      • 1970-01-01
      • 1970-01-01
      • 2018-01-26
      • 2017-11-14
      • 2018-06-25
      相关资源
      最近更新 更多