Build tensorflow使用外部库的新op

我写了一个从EXR文件读取数据的操作

代码可以编译，但是当我调用

tf.load\u op\u库（'custom\u ops.so'）

时，我有一个

NotFoundError:libcustom\u ops.so:undefined symbol:_ZNK3Imf9InputFile6headerEv

错误

我怀疑这是因为我的cc文件需要

libopenexr

库。我的问题是如何告诉tensorflow也加载其他库

另一种选择是，在my

custom_ops.so

中包含附加库

为此，我试着打电话给CMake

add_library(custom_ops SHARED ${SOURCES} back_warp.cu.o ${OpenEXR_LIBS})

---

但是它也有同样的错误。

我很抱歉。通过在CMakeLists.txt中添加以下内容，似乎可以解决此问题

target_link_libraries(custom_ops ${OpenEXR_LIBS})