使用tensorflow缺少库

piztneat  于 6个月前  发布在  其他
关注(0)|答案(1)|浏览(81)

我目前正在尝试使用深度学习工具deeplabcut。它曾经在我的机器上运行,但现在当我尝试训练数据集时,我得到错误:
NOT_FOUND:在检查点中未找到密钥efficientnet/efficientnet-b6/blocks_0/conv 2d/kernel
我试过安装不同版本的tensorflow,检查我的tensorflow,驱动程序,nvdia和conda环境都兼容。我甚至在不同的机器/GPU上重新安装了整个安装,使用python 3.9和3.8。错误消息总是相同的,表明检查点中有一些缺少的库。

sy5wg1nm

sy5wg1nm1#

它看起来像你做错了什么。https://github.com/tensorflow/tensorflow/issues/61468有它说这是不正确的,也许是一样的。
但是当不看的时候. https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/tf2tensorrt/utils/py_utils.cc
你的错误提示在第38行。那里说函数大约是GOOGLE_CUDA && GOOGLE_TENSORRT

bool IsGoogleTensorRTEnabled() {
#if GOOGLE_CUDA && GOOGLE_TENSORRT
#if TF_USE_TENSORRT_STATIC
  LOG(INFO) << "TensorRT libraries are statically linked, skip dlopen check";
  return true;
#else   // TF_USE_TENSORRT_STATIC
  auto handle_or = se::internal::DsoLoader::TryDlopenTensorRTLibraries();
  if (!handle_or.ok()) {
    LOG_WARNING_WITH_PREFIX << "Could not find TensorRT";
  }
  return handle_or.ok();
#endif  // TF_USE_TENSORRT_STATIC
#else   // GOOGLE_CUDA && GOOGLE_TENSORRT
  return false;
#endif  // GOOGLE_CUDA && GOOGLE_TENSORRT
}

字符串
希望这足够具体!

相关问题