那么,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.
可以用作起点的此类 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(使用前将#替换为@)