Tensorflow：如何安装 roi_pooling user_op

Question

我阅读了 post by HediBy that this ROI_POOLING user_op implementation works: LINK

我使用了 bazel build

-c opt //tensorflow/core/user_ops:roi_pooling.so to generate the so file (after installing tensorflow)

但是当我使用 tf.load_op_library to load roi_pooling.so 时，我得到了这个奇怪的错误：

 tensorflow.python.framework.errors.NotFoundError: /home/fishdrop/tensorflow/bazel-bin/tensorflow/core/user_ops/roi_pooling.so: undefined symbol: _Z21ROIPoolForwardLaucherPKffiiiiiiS0_PfPiRKN5Eigen9GpuDeviceE

有什么想法吗？ user_op 有其他人成功过吗？

Answer 1

感谢您的提示，我刚刚发现在没有 GPU 支持的情况下构建 user_op 时发生了错误。我重新安装了 tensorflow v 0.10，支持 GPU，并将所有 user_op 文件放在 //tensorflow/core/us‌ er_ops 中。

如果我使用 bazel build -c opt --config=cuda //tensorflow/core/us‌ er_ops:roi_pooling.so 编译 user_op（添加 -- config=cuda 不在 tensorflow user_op 文档中），user_op 现在可以工作了。我想这个问题现在可以关闭

Tensorflow：如何安装 roi_pooling user_op

Tensorflow: how to install roi_pooling user_op

python

tensorflow

detection

roi