| FROM nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04 |
| ENV DEBIAN_FRONTEND=noninteractive |
| ENV PYTHONUNBUFFERED=1 |
| ENV MPLCONFIGDIR=/tmp/matplotlib |
| RUN apt-get update && apt-get install -y \ |
| python3.11 \ |
| python3.11-venv \ |
| python3-pip \ |
| git \ |
| software-properties-common \ |
| wget \ |
| && apt-get clean \ |
| && rm -rf /var/lib/apt/lists/* |
| RUN add-apt-repository -y -r ppa:jonathonf/ffmpeg-4 \ |
| && apt-get update \ |
| && apt-get install -y ffmpeg \ |
| && apt-get clean \ |
| && rm -rf /var/lib/apt/lists/* |
| WORKDIR /app |
| RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 |
| RUN pip3 install numpy matplotlib pillow gradio==3.38.0 opencv-python ffmpeg-python moviepy |
| RUN git clone https://github.com/facebookresearch/segment-anything-2.git |
| WORKDIR /app/segment-anything-2 |
| |
| RUN pip3 install -e ".[demo]" |
| WORKDIR /app/segment-anything-2/checkpoints |
| RUN ./download_ckpts.sh |
| WORKDIR /app |
| EXPOSE 7860 |
| ENV GRADIO_SERVER_NAME="0.0.0.0" |
| COPY app.py /app/app.py |
| COPY assets /app/assets |
| CMD ["python3", "-u", "app.py"] |
|
|
