File size: 1,687 Bytes
035df8d
e85ab2c
3c9c655
035df8d
 
 
 
 
 
 
 
 
46faf7d
 
89b7fb7
46faf7d
 
01aee65
3780a00
 
 
 
 
 
3c9c655
 
 
46faf7d
f5c4dff
3c9c655
4af0100
 
 
0e0db7c
 
4af0100
 
0e0db7c
cf95753
0e0db7c
0dd3195
 
0e0db7c
1bc5de5
0e0db7c
 
 
 
 
 
 
3c9c655
0e0db7c
 
 
7f81961
0e0db7c
 
ff5da18
6dd0a2c
3c9c655
6dd0a2c
3c9c655
6dd0a2c
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
54
55
56
57
58
59
60
61
62
63
64
65
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 777 $WORKDIR


COPY requirements.txt $WORKDIR/requirements.txt
COPY . .
RUN ls
RUN pwd
COPY ./oneformer/data/bpe_simple_vocab_16e6.txt.gz $WORKDIR/oneformer/data/bpe_simple_vocab_16e6.txt.gz

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

# ARG TORCH_CUDA_ARCH_LIST=7.5+PTX


# RUN pip install ninja

# USER root
# RUN chown -R user:user /usr
# RUN chmod -R 777 /usr
# RUN chown -R user:user $HOME
# RUN chmod -R 777 $HOME
# RUN chown -R user:user $WORKDIR
# RUN chmod -R 777 $WORKDIR

# USER user
# 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
# RUN sh deform_setup.sh

USER user

EXPOSE 7860

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