创建一个编译Tensorflow二进制文件的Dockerfile:SSE4.1,SSE4.2和AVX指令

Question

那么,docker的一个主要目的是轻松部署一个环境来测试软件吗？任何人都可以告诉我如何编译Tensorflow二进制文件使用:docker文件上的SSE4.1,SSE4.2？任何人都可以指向我这样做的docker文件吗？如果可能的话？

总之,有两个问题:

• 是否有可能有一个编译Tensorflow二进制文件的docker文件:SSE4.1,SSE4.2(以及GPU,我只发现了一个或另一个)
• 你能告诉我在哪里可以找到这样做的Docker文件或一个好的教程吗？

"这个问题的目的是避免以下情况:主机设置工作但Docker设置不起作用,因为Tensorflow没有以特定方式编译." 如下图所示.

Answer 1

可以用作起点的此类 Dockerfile 的工作示例如下：https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/docker（README.md详情请参阅）。

更准确地说，它是一组参数化的 Docker 文件，构建以parameterized_docker_build.sh. 在 Docker 中成功编译 TensorFlow 的命令示例是：

export TF_DOCKER_BUILD_IS_DEVEL=YES
export TF_DOCKER_BUILD_TYPE=CPU
export TF_DOCKER_BUILD_PYTHON_VERSION=PYTHON3
export TF_DOCKER_BUILD_DEVEL_BRANCH=master
tensorflow/tools/docker/parameterized_docker_build.sh

为了使用自定义标志构建 TensorFlow，用作TF_DOCKER_BUILD_IS_DEVEL=YES非开发 Docker 文件，只需从服务器下载预编译的 Docker 二进制文件。

TensorFlow 团队最近刚刚开始使用 AVX 构建开发 Docker 镜像。

对于SSE，请参阅此问题。您可以在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile.devel的本地副本中修改 bazel 命令行。

附言。对于使用自定义选项的非开发 TensorFlow 构建，您可以查看https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/ci_build。