Docker 中的 Tensorflow GPU“SSE4.1 指令”问题

【问题标题】：Tensorflow GPU "SSE4.1 instructions" issue inside a DockerDocker 中的 Tensorflow GPU“SSE4.1 指令”问题
【发布时间】：2021-01-04 09:14:46
【问题描述】：

我尝试在 docker 中运行 tensorflow 1.13.1（所需配置的镜像是 evariste/autodl:gpu-latest）。

docker 可以访问 RTX 2080 Ti GPU。

我收到以下错误：

2020-09-10 16:09:47.428460: F tensorflow/core/platform/cpu_feature_guard.cc:37] The TensorFlow library was compiled to use SSE4.1 instructions, but these aren't available on your machine.

【问题讨论】：

标签： python docker tensorflow gpu sse

【解决方案1】：

SSE4.1 是 CPU 支持的指令集，不是 GPU 支持的。因此，您需要检查您的 CPU 是否支持它；关于这个话题的更多讨论可以找到here。

【讨论】：

你是对的。我的 CPU（Intel Xeon Gold 6126）支持 SSE4.2 指令集。
上面的 docker 镜像是为支持 SSE4.1 而构建的吗？如果不是（并且通常 Tensorflow 是在没有 SSE4.1 支持的情况下构建的），那么您可以放心地忽略该警告。
我不知道。这就是日志所暗示的。我个人的目标是在 GPU 上运行代码。
没关系，根据日志，图像可能支持SSE4.1，但是由于您只关心GPU，因此可以忽略警告，无论如何这不是错误。
这似乎是一个真正的错误，因为当我尝试导入 tensorflow 时 Python 崩溃了。使用 tensorflow 1.5.1 运行时，我没有问题（因为这个版本确实支持 SSE4.2）。现在我被卡住了，因为我真的需要运行 tf 1.13.1