Tensorflow:如何安装roi_pooling user_op

Question

我在HediBy的这文章中读到，这个ROI_POOLING user_op实现是有效的：链接

我用bazel建造

-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时发生了错误。我在GPU支持下重新安装了tensorflow v0.10，并将所有user_op文件放在//tensorflow/core/us‌​er_ops中。

如果我使用bazel build user_op opt -config=cuda// tensorflow /core/us‌​er_ops:roi_ -c编译-c，那么(tensorflow user_op文档中没有添加--config=cuda )，那么user_op现在就可以工作了。我想这个问题现在可以结束了