参考博文:http://blog.csdn.net/tianrolin/article/details/71246472
Makefile
181行
# LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5改为
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
409行
# NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)改为
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
Makefile.config
5行解注释
USE_CUDNN := 1(服务器Docker里没有cudnn,所以我没弄)
25行解注释
CUSTOM_CXX := g++
87行解注释
WITH_PYTHON_LAYER := 1
90-91行添加
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib