# EndoGaussian-4D: Real-Time Deformable 4D Gaussian Splatting for Endoscopic Surgery
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive
ENV CUDA_HOME=/usr/local/cuda
ENV PATH="${CUDA_HOME}/bin:${PATH}"
ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"

RUN apt-get update && apt-get install -y --no-install-recommends \
    python3.10 python3.10-dev python3-pip python3.10-venv \
    git wget curl unzip cmake ninja-build \
    libboost-all-dev libcgal-dev libeigen3-dev libflann-dev \
    libfreeimage-dev libmetis-dev libgoogle-glog-dev \
    libgflags-dev libsqlite3-dev libglew-dev qtbase5-dev \
    libqt5opengl5-dev libatlas-base-dev libsuitesparse-dev \
    libceres-dev colmap \
    ffmpeg libavcodec-dev libavformat-dev libswscale-dev \
    libgl1-mesa-glx libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

RUN ln -sf /usr/bin/python3.10 /usr/bin/python && \
    python -m pip install --upgrade pip setuptools wheel

RUN pip install torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121

COPY requirements.txt /tmp/requirements.txt
RUN pip install -r /tmp/requirements.txt
RUN pip install gsplat==1.4.0
RUN pip install transformers accelerate

WORKDIR /app/endogaussian4d
COPY . .
ENTRYPOINT ["python"]