Spaces:
Running
on
T4
Running
on
T4
File size: 1,489 Bytes
035df8d 91f0bb6 3c9c655 035df8d 5c23f00 89b7fb7 5c23f00 01aee65 3780a00 3c9c655 5c23f00 311617e 3c9c655 cf95753 3c9c655 035df8d 3780a00 3c9c655 be0b089 0dd3195 a0d4fd2 0dd3195 3f01ea0 c0d4288 2685271 3c9c655 5c23f00 7f81961 5c23f00 3c9c655 dcaec5d 3c9c655 dcaec5d |
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 |
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
USER root
# ENV HOME=/home/user \
# PATH=/home/user/.local/bin:$PATH
ENV HOME=/home/root \
PATH=/home/root/.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 chown -R root:root $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
RUN pip install ninja
USER root
RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build install --user && cd ..
RUN sh deform_setup.sh
USER root
RUN sh deform_setup.sh
USER root
EXPOSE 7860
ENTRYPOINT ["python", "gradio_app.py"]
|