File size: 1,424 Bytes
035df8d
91f0bb6
3c9c655
035df8d
 
 
 
 
 
 
 
 
3780a00
e137278
89b7fb7
3780a00
 
01aee65
3780a00
 
 
 
 
 
3c9c655
 
 
e469299
311617e
3c9c655
cf95753
3c9c655
035df8d
3780a00
3c9c655
 
 
be0b089
 
80717b7
a0d4fd2
d881494
c53dc5d
b360236
2685271
 
3c9c655
3a97a52
3c9c655
3a97a52
3c9c655
3a97a52
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
CMD nvidia-smi

ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update && apt-get install -y \
        git \
        make build-essential libssl-dev zlib1g-dev \
        libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
        libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev  \
    	ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
		&& rm -rf /var/lib/apt/lists/*

RUN useradd -ms /bin/bash user
USER user

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

RUN curl https://pyenv.run | bash
ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
RUN pyenv install 3.8.15 && \
    pyenv global 3.8.15 && \
    pyenv rehash && \
    pip install --no-cache-dir --upgrade pip setuptools wheel

ENV WORKDIR=/code
WORKDIR $WORKDIR
RUN chown -R user:user $WORKDIR
RUN chmod -R 755 $WORKDIR


COPY requirements.txt $WORKDIR/requirements.txt
COPY oneformer $WORKDIR/oneformer
RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt

COPY . .

ARG TORCH_CUDA_ARCH_LIST=7.5+PTX

USER user
RUN pip install ninja
RUN chown -R user:user $HOME
RUN chmod -R 755 $HOME
RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..

RUN sh deform_setup.sh

# USER user

# EXPOSE 7860

# ENTRYPOINT ["python", "gradio_app.py"]