diff options
author | Amit Patankar <amitpatankar@google.com> | 2018-05-07 12:45:52 -0700 |
---|---|---|
committer | Gunhan Gulsoy <gunan@google.com> | 2018-05-07 12:45:52 -0700 |
commit | 96dd8c99688290f90b9826af516b2c12d280c0ee (patch) | |
tree | db9d15b77406503be413c1ec1b18b943a94f9bc2 /tensorflow/tools/docker | |
parent | 48d656f208a4296796033555603c1c2da29dc499 (diff) |
Building development docker files with AVX as well. (#19130)
Diffstat (limited to 'tensorflow/tools/docker')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel | 2 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel b/tensorflow/tools/docker/Dockerfile.devel index b9996395d0..406d134699 100644 --- a/tensorflow/tools/docker/Dockerfile.devel +++ b/tensorflow/tools/docker/Dockerfile.devel @@ -85,7 +85,7 @@ RUN git clone --branch=r1.8 --depth=1 https://github.com/tensorflow/tensorflow.g ENV CI_BUILD_PYTHON python RUN tensorflow/tools/ci_build/builds/configured CPU \ - bazel build -c opt --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \ + bazel build -c opt --copt=-mavx --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \ # For optimized builds appropriate for the hardware platform of your choosing, uncomment below... # For ivy-bridge or sandy-bridge # --copt=-march="ivybridge" \ diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 7e5e6ef2d5..2fe47f3356 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -98,7 +98,7 @@ ENV TF_CUDNN_VERSION=7 RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \ LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:${LD_LIBRARY_PATH} \ tensorflow/tools/ci_build/builds/configured GPU \ - bazel build -c opt --config=cuda \ + bazel build -c opt --copt=-mavx --config=cuda \ --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \ tensorflow/tools/pip_package:build_pip_package && \ rm /usr/local/cuda/lib64/stubs/libcuda.so.1 && \ |