|
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 |
|
|
|
|
|
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 --no-cache-dir -e . |
|
|
|
RUN pip install --no-cache-dir . |
|
|
|
|
|
RUN pip list |
|
|
|
CMD sh $HOME/app/startup.sh |
|
|
