我写了一个从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})
但它也会出现同样的错误。
答案 0 :(得分：2)
我很抱歉。通过将以下内容添加到CMakeLists.txt
，似乎可以解决问题target_link_libraries(custom_ops ${OpenEXR_LIBS})