在 Tensorflow 和 Caffe 中实现的 Faster-R-CNN 训练模型错误答案

【问题标题】：Error in training model for Faster-R-CNN implemented in Tensorflow and Caffe在 Tensorflow 和 Caffe 中实现的 Faster-R-CNN 训练模型错误
【发布时间】：2017-08-14 23:49:29
【问题描述】：

我正在使用以下 github 存储库，它在 tensorflow 中实现了一个称为 Faster R-CNN 的深度学习模型，我想：

https://github.com/smallcorgi/Faster-RCNN_TF

然后，我运行命令（我想只用 CPU 运行）：

python ./tools/demo.py --model VGGnet_fast_rcnn_iter_70000.ckpt

我得到错误：

Traceback (most recent call last):
  File "./tools/demo.py", line 11, in <module>
    from networks.factory import get_network
  File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/__init__.py", line 8, in <module>
    from .VGGnet_train import VGGnet_train
  File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/VGGnet_train.py", line 2, in <module>
    from networks.network import Network
  File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/network.py", line 3, in <module>
    import roi_pooling_layer.roi_pooling_op as roi_pool_op
  File "/home/brm17/Faster-RCNN_TF/tools/../lib/roi_pooling_layer/roi_pooling_op.py", line 5, in <module>
    _roi_pooling_module = tf.load_op_library(filename)
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/load_library.py", line 64, in load_op_library
    None, None, error_msg, error_code)
tensorflow.python.framework.errors_impl.NotFoundError: /home/brm17/Faster-RCNN_TF/tools/../lib/roi_pooling_layer/roi_pooling.so: undefined symbol: _Z22ROIPoolBackwardLaucherPKffiiiiiiiS0_PfPKiRKN5Eigen9GpuDeviceE

问题是什么，如何解决？

【问题讨论】：

问题是你没有按照仓库中的说明进行操作。
欢迎来到 StackOverflow。请删除标签caffe，因为它不相关。

标签： tensorflow deep-learning caffe

【解决方案1】：

首先，您需要安装 cuda（7.5 或 8.0）（link to install cuda-8.0）。然后，您需要进入 .bashrc 文件（位于您的家中）并添加以下两行：

导出 LD_LIBRARY_PATH=/usr/local/cuda/lib

导出 PATH=$PATH:/usr/local/cuda/bin

.bashrc 文件保存后，运行以下命令：

源码.bashrc

不要忘记验证这两个变量是否存在（只需键入 $LD_LIBRARY_PATH 和 $PATH）。如果没有，请转到路径以更改文件夹的名称（例如，我必须在第一个变量中将“lib”更改为“lib64”）。

然后，它应该工作:-)

【讨论】：