簡體   English   中英

制作可編譯Tensorflow二進制文件以使用的Dockerfile:SSE4.1,SSE4.2和AVX指令

[英]Make a Dockerfile that compiles a Tensorflow binary to use: SSE4.1, SSE4.2 and AVX instructions

因此,泊塢窗的目的之一就是輕松部署環境來測試軟件,對嗎? 有人可以告訴我如何在docker文件上編譯Tensorflow二進制文件以使用:SSE4.1,SSE4.2嗎? 誰能指出我這樣做的碼頭工人文件? 是否有可能?

總之,有兩個問題:

  • 是否可以使用一個編譯Tensorflow二進制文件的docker文件使用:SSE4.1,SSE4.2(和GPU,我只找到了一個)
  • 您能告訴我在哪里可以找到可以做到這一點的Docker文件,或者是一個不錯的教程?

“此問題的目的是避免出現以下情況:主機設置有效但docker設置無效的地方,因為Tensorflow不是以特定方式編譯的。” 如下圖所示。 在此處輸入圖片說明

可以將此類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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM