Как связать OpenFST с пользовательской операцией tensorflow?

Я пытаюсь скомпилировать пользовательскую операцию tensorflow, для которой требуется OpenFST (http://www.openfst.org/twiki/bin/view/FST/WebHome). Однако я сталкиваюсь с ошибками компиляции, когда запускаю:

import tensorflow as tf
decoder_op = tf.load_op_library('./libfst_decoder.so')

и получить ошибку неопределенного символа undefined symbol: _ZN3fst21ConvertToLegalCSymbolEPSs, поэтому он не может найти связанный объект.

Это файл CMake, который я использую:

cmake_minimum_required(VERSION 3.5)

execute_process(COMMAND python3 -c "import tensorflow; print(tensorflow.sysconfig.get_include())" OUTPUT_VARIABLE Tensorflow_INCLUDE_DIRS)

set (CMAKE_CXX_FLAGS "--std=c++11 -fPIC -O2 -D_GLIBCXX_USE_CXX11_ABI=0")

link_directories("/miniconda/envs/py36/lib/python3.6/site-packages/tensorflow")
include_directories(${Tensorflow_INCLUDE_DIRS})
include_directories("/usr/local/include")
add_library(fst_decoder SHARED fst_decoder.cc simple_decoder.cc)
target_link_libraries(fst_decoder -ltensorflow_framework -lfst -ldl -lm)

Есть ли какие-либо очевидные проблемы с файлом CMake для включения внешней библиотеки? У меня нет большого опыта компиляции C++.


tensorflow openfst
person PC9494    schedule 29.05.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Решение состоит в том, чтобы перекомпилировать OpenFST с: make CFLAGS='-D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11' CXXFLAGS='-D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11', поскольку тензорный поток использует D_GLIBCXX_USE_CXX11_ABI=0.

person PC9494    schedule 29.05.2019
comment
Обратите внимание, что TF предоставляет некоторую функцию для автоматического получения этих флагов (tensorflow.sysconfig.get_flags() или что-то в этом роде). - person Albert; 08.11.2019