diff --git a/tensorflow/Dockerfile b/tensorflow/Dockerfile index 9aaf541c0ab645f5589bbe9c7c88b859940cddc1..1ac997501adf8e5406d0df2bf02bb6caf0539f69 100644 --- a/tensorflow/Dockerfile +++ b/tensorflow/Dockerfile @@ -21,8 +21,7 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf && \ ldconfig -ARG BAZEL_VERSION -ENV BAZEL_VERSION ${BAZEL_VERSION:-0.26.1} +ARG BAZEL_VERSION=0.26.1 RUN mkdir -p /opt/bazel && \ wget -O /opt/bazel/installer.sh "https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh" && \ wget -O /opt/bazel/LICENSE.txt "https://raw.githubusercontent.com/bazelbuild/bazel/master/LICENSE" && \ @@ -30,8 +29,7 @@ RUN mkdir -p /opt/bazel && \ /opt/bazel/installer.sh && \ rm -f /opt/bazel/installer.sh -ARG TF_VERSION -ENV TF_VERSION ${TF_VERSION:-1.15.2} +ARG TF_VERSION=1.15.2 RUN git clone --recursive --branch v${TF_VERSION} https://github.com/tensorflow/tensorflow.git /opt/tensorflow/src COPY disable-enforced-bazel-build-mirror.patch /opt/tensorflow/patches/disable-enforced-bazel-build-mirror.patch