FROM nvidia/cuda:12.4.1-base-ubuntu22.04

RUN apt-get -y update \
    && apt-get -y upgrade \
    && apt-get --no-install-recommends -y install git python3 python3-pip tree \
    && rm -rf /var/lib/apt/lists/*

RUN python3 -m pip install --upgrade pip \
    && pip install --no-cache-dir torch torchvision torchaudio
    # pyyaml-include==1.4.1

RUN useradd -m -u 1000 user

USER user

ENV HOME=/home/user \
        PATH=/home/user/.local/bin:$PATH

WORKDIR $HOME/app

COPY --chown=user . $HOME/app

RUN git clone --branch hf_demo https://github.com/TRI-ML/vlm-evaluation.git

# RUN cd vlm-evaluation && pip3 install -U --no-cache-dir -e .
RUN cd vlm-evaluation && pip3 install --no-cache-dir -e .

RUN pip install --no-cache-dir .
# RUN pip install -U --no-cache-dir .

RUN pip list

CMD sh $HOME/app/startup.sh