我尝试按照以下链接安装 tensorflow 和 docker https://www.tensorflow.org/install/docker
我将 tensorflow 图像从 tensorflow Hub 下载到 docker 中,然后我尝试测试 tensorflow python 脚本。然后我在下面收到错误消息。我认为它与 GPU 相关,但我刚刚下载了 tensorflow/tensorflow:last 并且我不需要 GPU 版本。我想如果我使用 docker + tensorflow 我不会得到任何错误。有没有人能告诉我这有什么问题。。
2020-02-15 08:24:32.759681: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory
2020-02-15 08:24:32.759786: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: cannot open shared object file: No such file or directory
2020-02-15 08:24:32.759798: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
Run Code Online (Sandbox Code Playgroud)
正如https://www.tensorflow.org/install/docker所说,“nvidia-container-runtime 仅适用于 Linux”。就目前而言,如果您使用 Windows 作为主机系统,则无法为 docker 容器提供 GPU。