python版本:python 3.6
tensorflow版本:tf 1.8.0,使用二进制pip软件包在网站上安装
cuda版本:9.0
cudnn版本:7.0
bazel版本:0.11.0
我使用以下命令编译了该库:
bazel build -c opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --local_resources=6144,4,1.0 --jobs=4 //tensorflow:libtensorflow_cc.so
当我在python中加载libtensorflow_cc.so
tf.load_op_library("libtensorflow_cc.so")
未定义的符号:_ZN9perftools8gputools6Stream15ThenRnnBackwardERKNS0_3dnn13RnnDescriptorERKNS2_27RnnSequenceTensorDescriptorERKNS0_12DeviceMemoryIdEERKNS2_24RnnStateTensorDescriptorESC_SF_SC_SC_S8_SC_SF_SC_SF_SC_SC_SC_SC_PSA_SG_SG_SG_PNS9_IhEEPNS0_16ScratchAllocatorE
deng @ xxx:〜$ c++filt _ZN9perftools8gputools6Stream15ThenRnnBackwardERKNS0_3dnn13RnnDescriptorERKNS2_27RnnSequenceTensorDescriptorERKNS0_12DeviceMemoryIdEERKNS2_24RnnStateTensorDescriptorESC_SF_SC_SC_S8_SC_SF_SC_SF_SC_SC_SC_SC_PSA_SG_SG_SG_PNS9_IhEEPNS0_16ScratchAllocatorEPNS2_13ProfileResultE
perftools :: gputools :: Stream :: ThenRnnBackward(perftools :: gputools :: dnn :: RnnDescriptor const&,perftools :: gputools :: dnn :: RnnSequenceTensorDescriptor const&,perftools :: gputools :: DeviceMemory const&, dnn :: RnnStateTensorDescriptor const&,perftools :: gputools :: DeviceMemory const&,perftools :: gputools :: dnn :: RnnStateTensorDescriptor const&,perftools :: gputools :: DeviceMemory const&,perftools :: gputools :: Device :dnn :: RnnSequenceTensorDescriptor const&,perftools :: gputools :: DeviceMemory const&,perftools :: gputools :: dnn :: RnnStateTensorDescriptor const&,perftools :: gputools :: DeviceMemory const&,perftools :: Gputools :: gputools :: :: gputools :: DeviceMemory const&,perftools :: gputools :: DeviceMemory const&,perftools :: gputools :: DeviceMemory const&,perftools :: gputools :: DeviceMemory ,perftools :: gputools :: DeviceMemory ,perftools :: gputools :: DeviceMemory ,perftools :: gputools :: DeviceMemory ,perftools :: gputools :: DeviceMemory ,perftools :: gputools :: ScratchAllocator )
我是否错过了定义perftools :: gputools :: Stream :: ThenRnnBackward(...)的库?我该如何解决?
我不知道cuda中是否有未定义的符号,我已经检查了libcudnn.so.7和cuda-9.0 / lib64中的库,没有这样的符号。
有人有什么主意吗?预先感谢。