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})
,但它給出了同樣的錯誤。