diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-15 17:32:50 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-15 17:39:26 -0800 |
commit | 9648f8040a559f6cf9bbe0501ba96f2b2c2864b1 (patch) | |
tree | 57dc6e959e0a534622eaf392ee43b7691378b10e /tensorflow/tools/docker | |
parent | 5b5445b9a7aa2664a90c4fc946ecf268c971425b (diff) |
Automated g4 rollback of changelist 179258973
PiperOrigin-RevId: 179260538
Diffstat (limited to 'tensorflow/tools/docker')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel | 5 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 26 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 2 | ||||
-rwxr-xr-x | tensorflow/tools/docker/parameterized_docker_build.sh | 2 |
4 files changed, 16 insertions, 19 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel b/tensorflow/tools/docker/Dockerfile.devel index 0a6860e791..3525c7524f 100644 --- a/tensorflow/tools/docker/Dockerfile.devel +++ b/tensorflow/tools/docker/Dockerfile.devel @@ -69,8 +69,11 @@ RUN mkdir /bazel && \ rm -f /bazel/bazel-$BAZEL_VERSION-installer-linux-x86_64.sh # Download and build TensorFlow. + +RUN git clone https://github.com/tensorflow/tensorflow.git && \ + cd tensorflow && \ + git checkout r1.4 WORKDIR /tensorflow -RUN git clone --branch=r1.4 --depth=1 https://github.com/tensorflow/tensorflow.git . # TODO(craigcitro): Don't install the pip package, since it makes it # more difficult to experiment with local changes. Instead, just add diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 4164cc3f88..041f45971b 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -1,20 +1,11 @@ -FROM nvidia/cuda:9.0-base-ubuntu16.04 +FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 LABEL maintainer="Craig Citro <craigcitro@google.com>" RUN apt-get update && apt-get install -y --no-install-recommends \ build-essential \ - cuda-command-line-tools-9-0 \ - cuda-cublas-dev-9-0 \ - cuda-cudart-dev-9-0 \ - cuda-cufft-dev-9-0 \ - cuda-curand-dev-9-0 \ - cuda-cusolver-dev-9-0 \ - cuda-cusparse-dev-9-0 \ curl \ git \ - libcudnn7=7.0.5.15-1+cuda9.0 \ - libcudnn7-dev=7.0.5.15-1+cuda9.0 \ libcurl3-dev \ libfreetype6-dev \ libpng12-dev \ @@ -26,11 +17,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ unzip \ zip \ zlib1g-dev \ + openjdk-8-jdk \ + openjdk-8-jre-headless \ wget \ && \ - rm -rf /var/lib/apt/lists/* && \ - find /usr/local/cuda-9.0/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ - rm /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a + apt-get clean && \ + rm -rf /var/lib/apt/lists/* RUN curl -fSsL -O https://bootstrap.pypa.io/get-pip.py && \ python get-pip.py && \ @@ -78,16 +70,18 @@ RUN mkdir /bazel && \ rm -f /bazel/bazel-$BAZEL_VERSION-installer-linux-x86_64.sh # Download and build TensorFlow. + +RUN git clone https://github.com/tensorflow/tensorflow.git && \ + cd tensorflow && \ + git checkout r1.4 WORKDIR /tensorflow -RUN git clone --branch=r1.4 --depth=1 https://github.com/tensorflow/tensorflow.git . # Configure the build for our CUDA configuration. ENV CI_BUILD_PYTHON python ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH ENV TF_NEED_CUDA 1 ENV TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.0,6.1 -ENV TF_CUDA_VERSION=9.0 -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} \ diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index b6682cd681..e212d10290 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 +FROM nvidia/cuda:8.0-cudnn6-runtime-ubuntu16.04 LABEL maintainer="Craig Citro <craigcitro@google.com>" diff --git a/tensorflow/tools/docker/parameterized_docker_build.sh b/tensorflow/tools/docker/parameterized_docker_build.sh index e7de7df856..80a07b9b3b 100755 --- a/tensorflow/tools/docker/parameterized_docker_build.sh +++ b/tensorflow/tools/docker/parameterized_docker_build.sh @@ -265,7 +265,7 @@ else DOCKERFILE="${TMP_DIR}/Dockerfile" # Modify the devel Dockerfile to specify the git branch - sed "s/^RUN git clone --branch=.* --depth=1/RUN git clone --branch=${TF_DOCKER_BUILD_DEVEL_BRANCH} --depth=1/" \ + sed -r "s/([\s]*git checkout )(.*)/\1${TF_DOCKER_BUILD_DEVEL_BRANCH}/g" \ "${ORIG_DOCKERFILE}" > "${DOCKERFILE}" # Modify python/pip version if necessary. |