diff options
author | 2016-06-30 01:11:12 -0700 | |
---|---|---|
committer | 2016-06-30 08:10:25 -0700 | |
commit | 30fc952faa4b100bc19ec1a7009e55c55a1b949b (patch) | |
tree | 71e70bb6faf5436aa4a5ad4ba1eb699ddcf683a7 /tensorflow/tools | |
parent | 125a96af8d0f2585c13f580ec4e8eb7d845fa3a9 (diff) | |
parent | 896086b24f0431e18af378a0d71d0494c221b356 (diff) |
Merging r0.9 back to master
Diffstat (limited to 'tensorflow/tools')
-rwxr-xr-x | tensorflow/tools/ci_build/builds/test_installation.sh | 1 | ||||
-rw-r--r-- | tensorflow/tools/dist_test/Dockerfile | 2 | ||||
-rw-r--r-- | tensorflow/tools/dist_test/server/Dockerfile | 2 | ||||
-rw-r--r-- | tensorflow/tools/dist_test/server/Dockerfile.test | 2 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile | 2 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 2 | ||||
-rw-r--r-- | tensorflow/tools/docker/README.md | 8 | ||||
-rwxr-xr-x | tensorflow/tools/docker/docker_run_gpu.sh | 13 | ||||
-rwxr-xr-x | tensorflow/tools/docker/parameterized_docker_build.sh | 3 | ||||
-rw-r--r-- | tensorflow/tools/pip_package/BUILD | 6 | ||||
-rw-r--r-- | tensorflow/tools/pip_package/setup.py | 2 |
11 files changed, 33 insertions, 10 deletions
diff --git a/tensorflow/tools/ci_build/builds/test_installation.sh b/tensorflow/tools/ci_build/builds/test_installation.sh index 173e3b254e..6fb40c7b65 100755 --- a/tensorflow/tools/ci_build/builds/test_installation.sh +++ b/tensorflow/tools/ci_build/builds/test_installation.sh @@ -89,6 +89,7 @@ PY_TEST_BLACKLIST="${PY_TEST_BLACKLIST}:"\ "tensorflow/contrib/quantization/python/dequantize_op_test.py:"\ "tensorflow/contrib/quantization/python/quantized_conv_ops_test.py:"\ "tensorflow/contrib/quantization/tools/quantize_graph_test.py:"\ +"tensorflow/contrib/session_bundle/exporter_test.py:"\ "tensorflow/python/platform/default/_resource_loader_test.py:"\ "tensorflow/python/platform/default/flags_test.py:"\ "tensorflow/python/platform/default/logging_test.py:"\ diff --git a/tensorflow/tools/dist_test/Dockerfile b/tensorflow/tools/dist_test/Dockerfile index b55433b310..5dc3e5565c 100644 --- a/tensorflow/tools/dist_test/Dockerfile +++ b/tensorflow/tools/dist_test/Dockerfile @@ -20,7 +20,7 @@ RUN /var/gcloud/google-cloud-sdk/bin/gcloud components install kubectl # Install nightly TensorFlow pip # TODO(cais): Should we build it locally instead? RUN pip install \ - http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.8.0-cp27-none-linux_x86_64.whl + http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.9.0-cp27-none-linux_x86_64.whl # Copy test files COPY scripts /var/tf-dist-test/scripts diff --git a/tensorflow/tools/dist_test/server/Dockerfile b/tensorflow/tools/dist_test/server/Dockerfile index 4cbc2f0645..c3bf751735 100644 --- a/tensorflow/tools/dist_test/server/Dockerfile +++ b/tensorflow/tools/dist_test/server/Dockerfile @@ -36,7 +36,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \ # Install TensorFlow CPU version from nightly build RUN pip --no-cache-dir install \ - http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.8.0-cp27-none-linux_x86_64.whl + http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.9.0-cp27-none-linux_x86_64.whl # Copy files, including the GRPC server binary at # server/grpc_tensorflow_server.py diff --git a/tensorflow/tools/dist_test/server/Dockerfile.test b/tensorflow/tools/dist_test/server/Dockerfile.test index af795e8066..de4411a05c 100644 --- a/tensorflow/tools/dist_test/server/Dockerfile.test +++ b/tensorflow/tools/dist_test/server/Dockerfile.test @@ -42,7 +42,7 @@ RUN pip install --upgrade pandas==0.18.1 # Install TensorFlow CPU version. RUN pip --no-cache-dir install \ - http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.8.0-cp27-none-linux_x86_64.whl + http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.9.0-cp27-none-linux_x86_64.whl # Copy files, including the GRPC server binary at # server/grpc_tensorflow_server.py diff --git a/tensorflow/tools/docker/Dockerfile b/tensorflow/tools/docker/Dockerfile index 19d27a50c9..1645d3f7f9 100644 --- a/tensorflow/tools/docker/Dockerfile +++ b/tensorflow/tools/docker/Dockerfile @@ -27,7 +27,7 @@ RUN pip --no-cache-dir install \ && \ python -m ipykernel.kernelspec -ENV TENSORFLOW_VERSION 0.8.0 +ENV TENSORFLOW_VERSION 0.9.0 # --- DO NOT EDIT OR DELETE BETWEEN THE LINES --- # # These lines will be edited automatically by parameterized_docker_build.sh. # diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index c6507c5e87..7f0765b4e8 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -27,7 +27,7 @@ RUN pip --no-cache-dir install \ && \ python -m ipykernel.kernelspec -ENV TENSORFLOW_VERSION 0.8.0 +ENV TENSORFLOW_VERSION 0.9.0 # --- DO NOT EDIT OR DELETE BETWEEN THE LINES --- # # These lines will be edited automatically by parameterized_docker_build.sh. # diff --git a/tensorflow/tools/docker/README.md b/tensorflow/tools/docker/README.md index 35bd03c62d..921d4e5353 100644 --- a/tensorflow/tools/docker/README.md +++ b/tensorflow/tools/docker/README.md @@ -31,7 +31,7 @@ Run non-GPU container using $ docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow -For GPU support install Nvidia drivers (ideally latest) and +For GPU support install NVidia drivers (ideally latest) and [nvidia-docker](https://github.com/NVIDIA/nvidia-docker). Run using $ nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu @@ -46,6 +46,12 @@ it there please and try using the nvidia-docker as described above. $ docker run -it -p 8888:8888 $CUDA_SO $DEVICES gcr.io/tensorflow/tensorflow:latest-gpu +## More containers + +See all available [tags](https://hub.docker.com/r/tensorflow/tensorflow/tags/) +for additional containers like release candidates or nighlty builds. + + ## Rebuilding the containers Just pick the dockerfile corresponding to the container you want to build, and run diff --git a/tensorflow/tools/docker/docker_run_gpu.sh b/tensorflow/tools/docker/docker_run_gpu.sh index 08f391ddf9..82c550b878 100755 --- a/tensorflow/tools/docker/docker_run_gpu.sh +++ b/tensorflow/tools/docker/docker_run_gpu.sh @@ -16,6 +16,19 @@ set -e +echo "===== DEPRECATED ===== DEPRECATED ===== DEPRECATED ===== DEPRECATED =====" +echo "DEPRECATED: Please use https://github.com/NVIDIA/nvidia-docker instead!" +echo " It does pretty much the same thing but better and it is" +echo " supported by NVidia." +echo "=========================================================================" + +export CUDA_HOME=${CUDA_HOME:-/usr/local/cuda} + +if [ ! -d ${CUDA_HOME}/lib64 ]; then + echo "Failed to locate CUDA libs at ${CUDA_HOME}/lib64." + exit 1 +fi + export CUDA_SO=$(\ls /usr/lib/x86_64-linux-gnu/libcuda.* | \ xargs -I{} echo '-v {}:{}') export DEVICES=$(\ls /dev/nvidia* | \ diff --git a/tensorflow/tools/docker/parameterized_docker_build.sh b/tensorflow/tools/docker/parameterized_docker_build.sh index dd4680bc0d..56770bed64 100755 --- a/tensorflow/tools/docker/parameterized_docker_build.sh +++ b/tensorflow/tools/docker/parameterized_docker_build.sh @@ -179,6 +179,9 @@ if [[ "${DO_PIP_BUILD}" == "1" ]]; then export TF_BUILD_IS_OPT="OPT" export TF_BUILD_IS_PIP="PIP" + export TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS=\ +"-e TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2" + pushd "${SCRIPT_DIR}/../../../" rm -rf pip_test/whl && tensorflow/tools/ci_build/ci_parameterized_build.sh diff --git a/tensorflow/tools/pip_package/BUILD b/tensorflow/tools/pip_package/BUILD index cf83a767e7..ae63181258 100644 --- a/tensorflow/tools/pip_package/BUILD +++ b/tensorflow/tools/pip_package/BUILD @@ -30,9 +30,9 @@ sh_binary( ":other_headers", ":simple_console", "//tensorflow:tensorflow_py", - "//tensorflow/contrib/session_bundle:all_files", - "//tensorflow/contrib/session_bundle:manifest_proto_py", - "//tensorflow/contrib/session_bundle/example:half_plus_two", +# "//tensorflow/contrib/session_bundle:all_files", +# "//tensorflow/contrib/session_bundle:manifest_proto_py", +# "//tensorflow/contrib/session_bundle/example:half_plus_two", "//tensorflow/contrib/slim:all_files", "//tensorflow/contrib/slim/python/slim/data:all_files", "//tensorflow/core:framework_headers", diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index 07b10de08c..59cb647e45 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -27,7 +27,7 @@ from setuptools import find_packages, setup, Command, Extension from setuptools.command.install import install as InstallCommandBase from setuptools.dist import Distribution -_VERSION = '0.8.0' +_VERSION = '0.9.0' numpy_version = "1.8.2" if platform.system() == "Darwin": |