我编写了一个从 EXR 文件读取数据的操作。
代码能够编译,但是当我调用tf.load_op_library('custom_ops.so')
时,我有一个NotFoundError: libcustom_ops.so: undefined symbol: _ZNK3Imf9InputFile6headerEv
错误。
我怀疑这是因为我的 cc 文件需要 libopenexr
库。我的问题是如何告诉 tensorflow 加载附加库?
另一种方法可能是在我的 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})
关于build - 使用外部库的tensorflow新操作,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42141960/