Upload Dockerfile

Dockerfile

@@ -1,11 +1,27 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
 
 # Basic deps
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && apt-get install -y \
-    python3.10 python3.10-venv python3-pip git curl ffmpeg colmap unzip \
+    python3.10 python3.10-venv python3-pip git curl ffmpeg unzip \
+    build-essential cmake libboost-all-dev libeigen3-dev libflann-dev \
+    libfreeimage-dev libmetis-dev libgoogle-glog-dev libgflags-dev \
+    libsqlite3-dev libglew-dev qtbase5-dev libqt5opengl5-dev libcgal-dev \
+    libceres-dev libsuitesparse-dev libgflags-dev libgoogle-glog-dev \
+    libatlas-base-dev libeigen3-dev libsuitesparse-dev \
  && rm -rf /var/lib/apt/lists/*
 
+# Install COLMAP from source
+RUN git clone https://github.com/colmap/colmap.git /tmp/colmap && \
+    cd /tmp/colmap && \
+    git checkout 3.8 && \
+    mkdir build && cd build && \
+    cmake .. -DCMAKE_BUILD_TYPE=Release && \
+    make -j$(nproc) && \
+    make install && \
+    ldconfig && \
+    rm -rf /tmp/colmap
+
 # Create venv and install nerfstudio + api deps
 RUN python3.10 -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"