FROM --platform=linux/amd64 nvidia/cuda:12.4.1-cudnn-devel-ubuntu20.04

ARG DEBIAN_FRONTEND=noninteractive

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

wget \

build-essential \

libvulkan1 \

mesa-vulkan-drivers \

xserver-xorg-core \

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

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 tk-dev uuid-dev zlib1g-dev

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

libjpeg-turbo8-dev \

libturbojpeg0-dev \

RUN wget https://www.python.org/ftp/python/3.11.9/Python-3.11.9.tgz

RUN tar -xvzf ./Python-3.11.9

WORKDIR Python-3.11.9/

RUN ./configure --enable-optimizations

RUN make -j 7

RUN altinstall

WORKDIR $HOME

RUN python3 -m venv ./venv

RUN source ./venv/bin/activate

RUN pip install setuptools --upgrade && pip install --upgrade pip

RUN pip install setuptools wheel

RUN pip install simplejpeg jupyterlab jupyter-rfb pygfx

RUN git clone https://github.com/fastplotlib/fastplotlib.git

WORKDIR fastplotlib/

RUN pip install -e .

ENTRYPOINT [ "/bin/bash" ]