Tensorflow: как установить roi_pooling user_op

В этом сообщении 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?


python tensorflow detection roi
person Link L    schedule 16.08.2016    source источник
comment
Вы пытаетесь запустить на GPU? Ваша командная строка сборки bazel неполная, но если вы пытаетесь запустить на графическом процессоре, я бы попробовал скомпилировать следующую командную строку: bazel build --config=cuda -c opt //tensorflow/core/user_ops:roi_pooling.so   -  person keveman    schedule 16.08.2016

Ответы (1)


arrow_upward
1
arrow_downward

спасибо за подсказку, я только что узнал, что ошибка произошла, когда я создал user_op без поддержки графического процессора. Я переустановил tensorflow v 0.10 с поддержкой графического процессора и поместил все файлы user_op внутрь //tensorflow/core/us‌​er_ops.

Если я скомпилирую user_op с помощью bazel build -c opt --config=cuda //tensorflow/core/us‌​er_ops:roi_pooling.so (добавление --config=cuda отсутствует в документации по tensorflow user_op), user_op теперь работает .. Думаю, теперь этот вопрос можно закрыть

person Link L    schedule 17.08.2016