diff options
Diffstat (limited to 'docker/Dockerfile')
| -rw-r--r-- | docker/Dockerfile | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/docker/Dockerfile b/docker/Dockerfile new file mode 100644 index 0000000..4481565 --- /dev/null +++ b/docker/Dockerfile @@ -0,0 +1,26 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +FROM quay.io/pypa/manylinux2014_x86_64 + +ARG PYTHON_VERSION=3.9 +ARG TENSORFLOW_VERSION=2.9.0 +ARG BAZEL_VERSION=5.1.1 + +RUN ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/python3 /usr/local/bin/python3 && \ + ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/python /usr/local/bin/python && \ + ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/pip3 /usr/local/bin/pip3 && \ + ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/pip /usr/local/bin/pip && \ + ln -s /opt/_internal/cpython-$PYTHON_VERSION*/include/python${PYTHON_VERSION}/ /usr/local/include/python + +ENV PYTHON_BIN_PATH=/usr/local/bin/python +ENV CI_BUILD_PYTHON=/usr/local/bin/python +ENV CROSSTOOL_PYTHON_INCLUDE_PATH=/usr/local/include/python/ + +RUN pip install --no-cache-dir setuptools pybind11 numpy twine keyrings.alt + +COPY install/install_bazel.sh /install/ +RUN /install/install_bazel.sh ${BAZEL_VERSION} + +# CACHE_STOP is used to rerun future commands, otherwise the cloning will be cached and will not pull the most recent version +ARG CACHE_STOP=1 +RUN git clone --depth=1 https://github.com/tensorflow/tensorflow.git --branch v${TENSORFLOW_VERSION} /tensorflow_src |