[英]Make a Dockerfile that compiles a Tensorflow binary to use: SSE4.1, SSE4.2 and AVX instructions
可以將此類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映像 。
對於上證所, 請參閱此問題 。 您可以在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile.devel的本地副本中修改bazel命令行。
PS。 對於具有自定義選項的非開發TensorFlow構建,您可以查看https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/ci_build 。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.