FROM nvidia/cuda:12.3.1-devel-ubuntu22.04

ARG USERNAME=vscode

COPY library-scripts/common-debian.sh /tmp/library-scripts/
RUN apt-get update && bash /tmp/library-scripts/common-debian.sh true automatic automatic automatic false true true


RUN SNIPPET="export PROMPT_COMMAND='history -a' && export HISTFILE=/commandhistory/.bash_history" \
    && mkdir -p /commandhistory \
    && touch /commandhistory/.bash_history \
    && chown -R $USERNAME /commandhistory \
    && echo "$SNIPPET" >> "/home/$USERNAME/.bashrc"

RUN SNIPPET="export PROMPT_COMMAND='history -a' && export HISTFILE=/commandhistory/.zsh_history" \
    && mkdir -p /commandhistory \
    && touch /commandhistory/.zsh_history \
    && chown -R $USERNAME /commandhistory \
    && echo "$SNIPPET" >> "/home/$USERNAME/.zshrc"

RUN apt-get update && export DEBIAN_FRONTEND=noninteractive \
    && apt-get -y install --no-install-recommends \
    vim \
    git \
    git-lfs \
    zsh \
    tmux \
    wget \
    python3 \
    python3-dev \
    python3-pip \
    python3-setuptools \
    libgl1-mesa-dev \
    libglib2.0-0 \
    unzip \
    curl

RUN git lfs install

USER vscode 

RUN python3 -m pip install --upgrade pip \
    && pip install --no-cache-dir \
    ultralytics \
    pillow \
    gradio

RUN curl -fsSL https://fnm.vercel.app/install | bash