Skip to content

Instantly share code, notes, and snippets.

@raghughanapuram
Last active April 22, 2018 14:04
Show Gist options
  • Save raghughanapuram/23dddc90fe3ea2d40b655bb8d00c0cff to your computer and use it in GitHub Desktop.
Save raghughanapuram/23dddc90fe3ea2d40b655bb8d00c0cff to your computer and use it in GitHub Desktop.
Undefined symbol in roi_pooling.so
python ./tools/demo.py --model ./lib/pretrained/VGGnet_fast_rcnn_iter_70000.ckpt
Std error:
tensorflow.python.framework.errors.NotFoundError: /home/dyc/workspace/TFFRCNN/lib/roi_pooling_layer/roi_pooling.so: undefined symbol: _ZN10tensorflow7strings6StrCatB5cxx11ERKNS0_8AlphaNumE
--------------------------------------------------------------------------------
Modify Faster-RCNN_TF/lib/make.sh as below:
# make.sh
TF_INC=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_include())')
TF_LIB=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')
CUDA_PATH=/usr/local/cuda-8.0/
CXXFLAGS=''
if [[ "$OSTYPE" =~ ^darwin ]]; then
CXXFLAGS+='-undefined dynamic_lookup'
fi
cd roi_pooling_layer
if [ -d "$CUDA_PATH" ]; then
nvcc -std=c++11 -c -o roi_pooling_op.cu.o roi_pooling_op_gpu.cu.cc \
-I $TF_INC -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC $CXXFLAGS \
-arch=sm_37
# g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc \
# roi_pooling_op.cu.o -I $TF_INC -D GOOGLE_CUDA=1 -fPIC $CXXFLAGS \
# -lcudart -L $CUDA_PATH/lib64
# g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc \
# roi_pooling_op.cu.o -I $TF_INC -D GOOGLE_CUDA=1 -fPIC $CXXFLAGS -D_GLIBCXX_USE_CXX11_ABI=0 \
# -lcudart -L $CUDA_PATH/lib64
g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc \
roi_pooling_op.cu.o -I $TF_INC -I $TF_INC/external/nsync/public \
-D GOOGLE_CUDA=1 -fPIC $CXXFLAGS -D_GLIBCXX_USE_CXX11_ABI=0 \
-L$TF_LIB -ltensorflow_framework -lcudart -L $CUDA_PATH/lib64
else
# g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc \
# -I $TF_INC -fPIC $CXXFLAGS
g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc \
-I $TF_INC -fPIC $CXXFLAGS -L$TF_LIB -ltensorflow_framework
fi
cd ..
#cd feature_extrapolating_layer
#nvcc -std=c++11 -c -o feature_extrapolating_op.cu.o feature_extrapolating_op_gpu.cu.cc \
# -I $TF_INC -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -arch=sm_50
#g++ -std=c++11 -shared -o feature_extrapolating.so feature_extrapolating_op.cc \
# feature_extrapolating_op.cu.o -I $TF_INC -fPIC -lcudart -L $CUDA_PATH/lib64
#cd ..
#________________________________________________________________________________________
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment