ARG BASE_IMAGE=debian

ARG BASE_IMAGE_TAG=13

ARG CUDA_IMAGE

ARG CUDA_IMAGE_SUBTAG

ARG BLAS=libopenblas-dev

ARG CUDA_VERSION

ARG PYTHON_VERSION

FROM glcr.b-data.ch/python/psi${PYTHON_VERSION:+/}${PYTHON_VERSION:-:none}${PYTHON_VERSION:+/$BASE_IMAGE}${PYTHON_VERSION:+:$BASE_IMAGE_TAG} as psi

FROM ${CUDA_IMAGE:-$BASE_IMAGE}:${CUDA_IMAGE:+$CUDA_VERSION}${CUDA_IMAGE:+-}${CUDA_IMAGE_SUBTAG:-$BASE_IMAGE_TAG}

ARG DEBIAN_FRONTEND=noninteractive

ARG BASE_IMAGE

ARG BASE_IMAGE_TAG

ARG CUDA_IMAGE

ARG CUDA_IMAGE_SUBTAG

ARG BLAS

ARG CUDA_VERSION

ARG PYTHON_VERSION

ARG BUILD_START

ARG CUDA_IMAGE_LICENSE=${CUDA_IMAGE:+"NVIDIA Deep Learning Container License"}

ARG IMAGE_LICENSE=${CUDA_IMAGE_LICENSE:-"MIT"}

ARG IMAGE_SOURCE=https://gitlab.b-data.ch/python/docker-stack

ARG IMAGE_VENDOR="b-data GmbH"

ARG IMAGE_AUTHORS="Olivier Benz <[email protected]>"

LABEL org.opencontainers.image.licenses="$IMAGE_LICENSE" \

org.opencontainers.image.source="$IMAGE_SOURCE" \

org.opencontainers.image.vendor="$IMAGE_VENDOR" \

org.opencontainers.image.authors="$IMAGE_AUTHORS"

ENV BASE_IMAGE=${BASE_IMAGE}:${BASE_IMAGE_TAG} \

CUDA_IMAGE=${CUDA_IMAGE}${CUDA_IMAGE:+:}${CUDA_IMAGE:+$CUDA_VERSION}${CUDA_IMAGE:+-}${CUDA_IMAGE_SUBTAG} \

PARENT_IMAGE=${CUDA_IMAGE:-$BASE_IMAGE}:${CUDA_IMAGE:+$CUDA_VERSION}${CUDA_IMAGE:+-}${CUDA_IMAGE_SUBTAG:-$BASE_IMAGE_TAG} \

PYTHON_VERSION=${PYTHON_VERSION} \

BUILD_DATE=${BUILD_START}

ENV LANG=en_US.UTF-8 \

TERM=xterm \

TZ=Etc/UTC

## Install Python

COPY --from=psi /usr/local /usr/local

RUN apt-get update \

&& apt-get install -y --no-install-recommends \

ca-certificates \

libexpat1-dev \

libffi-dev \

libgdbm-dev \

libjs-sphinxdoc \

liblapack-dev \

libncurses-dev \

${BLAS} \

libreadline-dev \

locales \

netbase \

pkg-config \

tzdata \

unzip \

zip \

zlib1g-dev \

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

## Install system Python

apt-get -y install --no-install-recommends \

python3; \

## make some useful symlinks that are expected to exist

## ("/usr/bin/python" and friends)

for src in pydoc3 python3 python3-config; do \

dst="$(echo "$src" | tr -d 3)"; \

if [ -s "/usr/bin/$src" ] && [ ! -e "/usr/bin/$dst" ]; then \

ln -svT "$src" "/usr/bin/$dst"; \

fi \

done; \

fi \

## Update locale

&& sed -i "s/# $LANG/$LANG/g" /etc/locale.gen \

&& locale-gen \

&& update-locale LANG=$LANG \

## Switch BLAS/LAPACK (manual mode)

&& if [ ${BLAS} = "libopenblas-dev" ]; then \

update-alternatives --set libblas.so.3-$(uname -m)-linux-gnu \

/usr/lib/$(uname -m)-linux-gnu/openblas-pthread/libblas.so.3; \

update-alternatives --set liblapack.so.3-$(uname -m)-linux-gnu \

/usr/lib/$(uname -m)-linux-gnu/openblas-pthread/liblapack.so.3; \

fi \

## Clean up

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

CMD ["python3"]