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