编译caffe2源码遇到的坑

编译caffe2源码遇到的坑

最新的caffe2 已经移到pytorch中去了，直接下载其中的代码然后按照官方教程就可以了。

编译遇到的问题：

make[3]: Entering directory ‘/home/ubuntu/caffe2-master/build’ 
make[3]: Leaving directory ‘/home/ubuntu/caffe2-master/build’ 
make[3]: Entering directory ‘/home/ubuntu/caffe2-master/build’ 
[ 92%] Linking CXX executable ../bin/mpi_test 
/usr/bin/ld: CMakeFiles/mpi_test.dir/mpi/mpi_test.cc.o: undefined reference to symbol ‘_ZN3MPI8Datatype4FreeEv’ 
//usr/lib/libmpi_cxx.so.1: error adding symbols: DSO missing from command line 
collect2: error: ld returned 1 exit status 
caffe2/CMakeFiles/mpi_test.dir/build.make? recipe for target ‘bin/mpi_test’ failed 
make[3]: * [bin/mpi_test] Error 1 
make[3]: Leaving directory ‘/home/ubuntu/caffe2-master/build’ 
CMakeFiles/Makefile2:2654: recipe for target ‘caffe2/CMakeFiles/mpi_test.dir/all’ failed 
make[2]: * [caffe2/CMakeFiles/mpi_test.dir/all] Error 2 
make[2]: Leaving directory ‘/home/ubuntu/caffe2-master/build’ 
Makefile:140: recipe for target ‘all’ failed 
make[1]: * [all] Error 2 
make[1]: Leaving directory ‘/home/ubuntu/caffe2-master/build’ 
Makefile:4: recipe for target ‘all’ failed 
make: * [all] Error 2

问题出在MPI的链接问题中。

因此你需要修改的文件如下： 
pytorch->caffe2->CMakefiles.txt 
添加如下到最后行就可以了： 
SET(CMAKE_C_COMPILER mpicc) 
SET(CMAKE_CXX_COMPILER mpicxx) 
include_directories(MPI_INCLUDE_PATH)

target_link_libraries(mpi_test ${MPI_LIBRARIES})

测试遇到的问题

再输入测试时候会遇到问题： 
python -c ‘from caffe2.python import core’ 
发现不能import caffe2_pb，这一步其实是因为路径存在问题。 
如果你位于PATH_TO/pytorch/build 
你就会发现，可以解决了，因此你需要把这个目录加载到PYTHONPATH中去。 
具体方法： 
echo ‘export PYTHONPATH=$PYTHONPATH:PATH_TO/pytorch/build’ >> ~/.bashrc中去。 
然后source ~/.bashrc就可以了。

其中PATH_TO需要根据你自己的电脑上的目录操作。