diff --git a/Dockerfile b/Dockerfile index 741f30c080269149254c75bbe410f276a97fd9ea..251f5958791824092046accd4d06a3fdc7cd6a14 100644 --- a/Dockerfile +++ b/Dockerfile @@ -40,15 +40,15 @@ WORKDIR /src/tf RUN git config --global advice.detachedHead false ### TF -ARG TF=v2.5.0 +ARG TF=v2.8.0 # Install bazelisk (will read .bazelversion and download the right bazel binary - latest by default) RUN wget -qO /opt/otbtf/bin/bazelisk https://github.com/bazelbuild/bazelisk/releases/latest/download/bazelisk-linux-amd64 \ && chmod +x /opt/otbtf/bin/bazelisk \ && ln -s /opt/otbtf/bin/bazelisk /opt/otbtf/bin/bazel -ARG BZL_TARGETS="//tensorflow:libtensorflow_cc.so //tensorflow/tools/pip_package:build_pip_package" +ARG BZL_TARGETS="//tensorflow/tools/lib_package:libtensorflow //tensorflow/tools/pip_package:build_pip_package" # "--config=opt" will enable 'march=native' (otherwise read comments about CPU compatibility and edit CC_OPT_FLAGS in build-env-tf.sh) -ARG BZL_CONFIGS="--config=nogcp --config=noaws --config=nohdfs --config=opt --config=monolithic" +ARG BZL_CONFIGS="--config=nogcp --config=noaws --config=nohdfs --config=opt --config=monolithic --config=v2" # "--compilation_mode opt" is already enabled by default (see tf repo .bazelrc and configure.py) ARG BZL_OPTIONS="--verbose_failures --remote_cache=http://localhost:9090"