首页 > 解决方案 > tensorflow/tensorflow-gpu:2.6.0 docker 镜像坏了?

问题描述

一段时间以来,我正在使用 tensorflow 在 GPU 上进行对象检测。我有以下问题:我被迫坚持使用 2.3.0 版本,因为 2.3.0 之后的所有版本都失败并出现以下错误:

Traceback (most recent call last):
  File "src/main/python/roldetector/video_rol_detector.py", line 6, in <module>
    import tensorflow as tf
  File "/home/tensorflow/.local/lib/python3.6/site-packages/tensorflow/__init__.py", line 438, in <module>
    _ll.load_library(_main_dir)
  File "/home/tensorflow/.local/lib/python3.6/site-packages/tensorflow/python/framework/load_library.py", line 154, in load_library
    py_tf.TF_LoadLibrary(lib)
tensorflow.python.framework.errors_impl.NotFoundError: /usr/local/lib/python3.6/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so: undefined symbol: _ZN10tensorflow8OpKernel11TraceStringEPNS_15OpKernelContextEb

我知道有更多的人遇到这个问题,但我还没有看到一个可行的解决方案。

最引人注目的——几乎是自相矛盾的——是 docker 应该防止这种问题。我正在使用 ubuntu 20.04,但我不需要指定这一点,因为如果 docker 映像是正确的,那么主机上运行的内容就无关紧要了。正确的?

有没有人有一个可行的解决方案,所以我可以开始使用最新版本的 tensorflow?(2.3.0 之后的所有版本似乎都遇到了同样的问题)

问候,克里斯

标签: dockertensorflow

解决方案


推荐阅读