2017-02-09 75 views
2

我寫了一個從EXR文件讀取數據的操作。tensorflow使用外部庫的新操作

代碼能夠編譯,但是當我打電話給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}) 

,但它給出了同樣的錯誤。

回答

2

非常抱歉。這個問題似乎可以通過向CMakeLists.txt添加以下內容來解決:

target_link_libraries(custom_ops ${OpenEXR_LIBS})