ARG CUDA_VERSION=12.2.0-devel-ubuntu22.04

FROM nvidia/cuda:${CUDA_VERSION} as cuda-devel-base
RUN apt-get update && \
    apt-get install -y make build-essential gdb lcov pkg-config \
        libbz2-dev libffi-dev libgdbm-dev libgdbm-compat-dev liblzma-dev \
        libncurses5-dev libreadline6-dev libsqlite3-dev libssl-dev \
        lzma lzma-dev uuid-dev zlib1g-dev

FROM cuda-devel-base as cuda-devel-python3.8-building
ADD ./Python-3.8.11.tgz /tmp/
WORKDIR /tmp/Python-3.8.11
RUN ./configure --prefix=/usr/local/python3.8 --enable-optimizations && \
    make && make install && \
    /usr/local/python3.8/bin/python3 -V && \
    /usr/local/python3.8/bin/pip3 -V

FROM cuda-devel-base
COPY --from=cuda-devel-python3.8-building /usr/local/python3.8 /usr/local/python3.8
RUN ln -sf /usr/local/python3.8/bin/python3 /usr/bin/python3 && \
    ln -sf /usr/local/python3.8/bin/pip3 /usr/bin/pip3