|
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"] |
|
|
