制作一个编译 Tensorflow 二进制文件的 Dockerfile 以使用:SSE4.1、SSE4.2 和 AVX 指令

2024-01-01

那么,docker的目的之一就是轻松部署一个环境来测试软件,对吧?谁能告诉我如何编译 Tensorflow 二进制文件以在 docker 文件上使用:SSE4.1、SSE4.2?有人能指点我一个可以做到这一点的 docker 文件吗?如果有可能吗?

总结来说,有两个问题:

  • 是否有可能有一个编译 Tensorflow 二进制文件以使用的 docker 文件:SSE4.1、SSE4.2(和 GPU,我只找到了其中之一)
  • 你能告诉我在哪里可以找到执行此操作的 docker 文件或好的教程吗?

"The purpose of this question is to avoid the following scenario: Where the host setup work but the docker set up doesn't work because Tensorflow was not compiled in a particular way." Like the image below.enter image description here


可以用作起点的此类 Dockerfile 的工作示例如下:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/docker https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/docker (see 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 镜像 https://github.com/tensorflow/tensorflow/commit/96dd8c99688290f90b9826af516b2c12d280c0ee最近。

For SSE 。您可以在本地副本中修改 bazel 命令行https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile.devel https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile.devel.

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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

制作一个编译 Tensorflow 二进制文件的 Dockerfile 以使用:SSE4.1、SSE4.2 和 AVX 指令 的相关文章

随机推荐