File size: 712 Bytes
a7119a1
54e6841
bc2999d
 
 
54e6841
 
bc2999d
4d27f4c
54e6841
3567bb7
 
 
 
 
bc2999d
3567bb7
 
 
 
54e6841
0ac9bc7
9e32bc9
9e5cbc3
9e32bc9
9e5cbc3
9e32bc9
f0edafb
 
1e6d6d0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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 pip install -U --no-cache-dir .

RUN pip list

CMD sh $HOME/app/startup.sh