Spaces:
Running
on
T4
Running
on
T4
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 | |
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 user | |
RUN sh deform_setup.sh | |
USER user | |
EXPOSE 7860 | |
ENTRYPOINT ["python", "gradio_app.py"] | |