ARG TF_CONTAINER_VERSION

FROM tensorflow/tensorflow:${TF_CONTAINER_VERSION}

LABEL version="1.0" maintainer="Jonathan DEKHTIAR <contact@jonathandekhtiar.eu>"

ARG TL_VERSION

ARG TF_CONTAINER_VERSION

RUN echo "Container Tag: ${TF_CONTAINER_VERSION}" \

&& apt-get update \

&& case $TF_CONTAINER_VERSION in \

latest-py3 | latest-gpu-py3) apt-get install -y python3-tk ;; \

*) apt-get install -y python-tk ;; \

esac \

&& if [ -z "$TL_VERSION" ]; then \

echo "Building a Nightly Release" \

&& apt-get install -y git \

&& mkdir /dist/ && cd /dist/ \

&& git clone https://github.com/tensorlayer/tensorlayer.git \

&& cd tensorlayer \

&& pip install --disable-pip-version-check --no-cache-dir --upgrade -e .[all]; \

else \

echo "Building Tag Release: $TL_VERSION" \

&& pip install --disable-pip-version-check --no-cache-dir --upgrade tensorlayer[all]=="$TL_VERSION"; \

fi \

&& apt-get autoremove -y \

&& rm -rf /var/lib/apt/lists/*