Dockerfile

FROM alpine:3.12

# Add Edge repos
RUN echo -e "\n\
@edgemain http://nl.alpinelinux.org/alpine/edge/main\n\
@edgecomm http://nl.alpinelinux.org/alpine/edge/community\n\
@edgetest http://nl.alpinelinux.org/alpine/edge/testing"\
  >> /etc/apk/repositories

ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin \
    LANG=zh_CN.UTF-8 \
    SHELL=/bin/bash PS1="\u@\h:\w \$ " \
    JAVA_HOME=/usr/lib/jvm/default-jvm \
    BAZEL_VERSION=0.25.0 \
    TENSORFLOW_VERSION=1.15.5 \
    EXTRA_BAZEL_ARGS=--host_javabase=@local_jdk//:jdk

RUN apk add --no-cache python3 python3-tkinter py3-numpy hdf5@edgetest py3-numpy-f2py freetype libpng libjpeg-turbo imagemagick graphviz git bash \
    && apk add --no-cache --virtual=.build-deps \
        coreutils \
        protobuf \
        cmake \
        curl \
        freetype-dev \
        g++ \
        libjpeg-turbo-dev \
        libpng-dev \
        libcurl \
        libstdc++ \
        linux-headers \
        make \
        musl-dev \
        openblas-dev \
        openjdk8 \
        patch \
        perl \
        python3-dev \
        py3-numpy-dev \
        rsync \
        sed \
        swig \
        zip \
        hdf5-dev@edgetest \
        libexecinfo-dev \
        && rm -rf /var/cache/apk/* \
        && cd /tmp \
        && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
        && python3 get-pip.py \
        && pip3 install --no-cache-dir numpy==1.18.0 h5py==2.9.0 \
        && pip3 install -U --user keras_preprocessing keras_applications --no-deps \
        && pip3 install --no-cache-dir setuptools wheel \
        && $(cd /usr/bin && ln -s python3 python) \
        && rm -f /tmp/get-pip.py
# Bazel download and install
RUN curl -SLO https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-dist.zip \
        && mkdir bazel-${BAZEL_VERSION} \
        && unzip -qd bazel-${BAZEL_VERSION} bazel-${BAZEL_VERSION}-dist.zip \
        && cd bazel-${BAZEL_VERSION} \
        && sed -i -e 's/-classpath/-J-Xmx8192m -J-Xms128m -classpath/g' scripts/bootstrap/compile.sh \
        && bash compile.sh \
        && cp -p output/bazel /usr/local/bin/ \
        && cp -p output/bazel /usr/bin/
# Download and Build Tensorflow
# 27285 total actions.In a 2.8G intel core and 4G memory , it will cost about one day. In 8 3.7G intel core , it will cost 4 hours. Github action will be timeout.
RUN cd /tmp \
    && curl -SL https://github.com/tensorflow/tensorflow/archive/v${TENSORFLOW_VERSION}.tar.gz \
        | tar xzf - \
    && : musl-libc error \
    && curl https://raw.githubusercontent.com/thy486/alpine-tensorflow/master/fix/env.cc -o env.cc \
    && cp -rf env.cc /tmp/tensorflow-${TENSORFLOW_VERSION}/tensorflow/core/platform/posix/env.cc \
    && cd /tmp/tensorflow-${TENSORFLOW_VERSION} \
    && sed -i -e '/undef HAVE_SYS_SYSCTL_H.*define HAVE_SYS_SYSCTL_H 1/d' third_party/hwloc/BUILD.bazel \
    && sed -i -e '/define TF_GENERATE_BACKTRACE/d' tensorflow/core/platform/default/stacktrace.h \
    && sed -i -e '/define TF_GENERATE_STACKTRACE/d' tensorflow/core/platform/default/stacktrace_handler.cc \
    && sed -i "s#nullptr.*/\* tp_print \*/#NULL, /\* tp_print \*/#g" tensorflow/python/lib/core/ndarray_tensor_bridge.cc \
    && sed -i "s#nullptr.*// tp_print#NULL, // tp_print#g" tensorflow/python/lib/core/bfloat16.cc \
    && sed -i "s#nullptr.*/\* tp_print \*/#NULL, /\* tp_print \*/#g" tensorflow/python/eager/pywrap_tfe_src.cc \
    && sed -i -e '/HAVE_BACKTRACE/d' third_party/llvm/llvm.bzl \
    && sed -i -e '/HAVE_MALLINFO/d' third_party/llvm/llvm.bzl \
    && rm -f /tmp/env.cc \
    && PYTHON_BIN_PATH=/usr/bin/python \
        PYTHON_LIB_PATH=/usr/lib/python3.8/site-packages \
        CC_OPT_FLAGS="-march=native" \
        TF_NEED_OPENCL=0 \
        TF_ENABLE_XLA=1 \
        TF_NEED_OPENCL_SYCL=0 \
        TF_NEED_S3=0 \
        TF_NEED_ROCM=0 \
        TF_NEED_CUDA=0 \
        TF_DOWNLOAD_CLANG=0 \
        TF_NEED_MPI=0 \
        TF_SET_ANDROID_WORKSPACE=0 \
        bash configure \
        && cd /tmp/tensorflow-${TENSORFLOW_VERSION} \
        && bazel build --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \
        # --config=mkl \
        --config=noaws \
        --config=nogcp \
        --config=nohdfs \
        # --config=nonccl \
        # --config=nokafka \
        # --config=noignite \
        -c opt \
        //tensorflow/tools/pip_package:build_pip_package
RUN cd /tmp/tensorflow-${TENSORFLOW_VERSION} \
        && ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg \
        && mkdir -p /root/tensorflow_pkg \
        && cp -rf /tmp/tensorflow_pkg/* /root/tensorflow_pkg/
        # && apk del .build-deps .build-deps.hdf5 \
        # && rm -rf /tmp/* /root/.cache

# Make sure it's built properly
RUN pip3 install --no-cache-dir /root/tensorflow_pkg/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl \
        && python3 -c 'import tensorflow; print("success")'