无法从caffe2.python导入caffe2?

时间:2018-01-01 14:24:04

标签: python python-2.7 anaconda caffe

我试图在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

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

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"

它不起作用。我该怎么办?

1 个答案:

答案 0 :(得分:-2)

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 build的引用:

# - mkl-include
# - mkl

运行:

# - mkl-include
# - mkl