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 user RUN chmod -R 755 $HOME/.pyenv RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py develop --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"]