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

Я прочитал в this сообщению HediBy, что эта реализация ROI_POOLING user_op работы: LINK Tensorflow: как установить 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?

источник

2016-08-16 Link L

Вы пытаетесь запустить на GPU? Ваша командная строка сборки bazel не завершена, но если вы пытаетесь запустить на GPU, я бы попытался выполнить компиляцию с помощью следующей командной строки: 'bazel build --config = cuda -c opt // tensorflow/core/user_ops: roi_pooling. so' – keveman

это работает спасибо –

спасибо за отзыв Я только узнал, что ошибка произошла, когда я создал user_op без поддержки графического процессора. Я переустановил shadoworflow v 0.10 с поддержкой GPU и разместил все файлы 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 теперь работает. Думаю, эта проблема теперь может быть закрыта

источник

2016-08-17 01:32:24

ответ

Смежные вопросы