Faster-RCNN运行时遇到的问题

在尝试运行Faster-RCNN的demo时遇到'roi_pooling_layer/roi_pooling.so: undefined symbol: _ZTIN10tensorflow8OpKernelE'错误。通过检查并修改make.sh文件,添加TF_LIB变量,调整g++编译配置,解决_GLIBCXX_USE_CXX11_ABI问题,更新CUDA路径,并修改CXXFLAGS,最终成功编译并运行demo。
摘要由CSDN通过智能技术生成

从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/ (

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值