从https://github.com/smallcorgi/Faster-RCNN_TF下载源码,根据说明下载各种文件
在运行demo的时候遇到错误
roi_pooling_layer/roi_pooling.so: undefined symbol: _ZTIN10tensorflow8OpKernelE
第一步到lib下查看make.sh文件,添加
TF_LIB=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')
第二步修改g++配置,如下所示:
g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc -D_GLIBCXX_USE_CXX11_ABI=0 \
roi_pooling_op.cu.o -I $TF_INC -L $TF_LIB -ltensorflow_framework -D GOOGLE_CUDA=1 \
-fPIC $CXXFLAGS -lcudart -L $CUDA_PATH/lib64
第三步,查看自己的CUDA路径,CUDA_PATH=/usr/local/cuda-9.0/,查看/usr/local是否存在cuda,根据自己的文件夹名字修改路径,随后在终端中执行export PATH=$PATH:/usr/local/cuda-9.0/bin/ (