packages and repo for ppsurf

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && \
     apt-get upgrade -y && \
@@ -7,7 +7,6 @@ RUN apt-get update && \
     wget \
     curl \
     zip \
-    # ffmpeg \
     ffmpeg \
     x264 \
     # python build dependencies \
@@ -35,22 +34,19 @@ WORKDIR ${HOME}/app
 
 RUN curl https://pyenv.run | bash
 ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
-ARG PYTHON_VERSION=3.10.11
+ARG PYTHON_VERSION=3.11.8
 RUN pyenv install ${PYTHON_VERSION} && \
     pyenv global ${PYTHON_VERSION} && \
     pyenv rehash && \
     pip install --no-cache-dir -U pip setuptools wheel
 
-RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir -U torch==2.2.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
 COPY --chown=1000 requirements.txt /tmp
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
-RUN pip install --no-cache-dir -U kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.1.1_cu121.html
-# RUN pip install --no-cache-dir -U gradio==4.19.1
 
 COPY --chown=1000 . ${HOME}/app
-RUN git clone https://github.com/TEXTurePaper/TEXTurePaper.git
-RUN cd TEXTurePaper && patch -p1 < ../patch
-# RUN cd ppsurf && patch -p1 < ../patch
+RUN git clone git@github.com:cg-tuwien/ppsurf.git
+RUN cd ppsurf
 ENV PYTHONPATH=${HOME}/app \
     PYTHONUNBUFFERED=1 \
     GRADIO_ALLOW_FLAGGING=never \

requirements.txt

@@ -1,18 +1,25 @@
-accelerate==0.19.0
-diffusers==0.16.1
-einops==0.6.1
-huggingface-hub==0.19.3
-imageio[ffmpeg,pyav]==2.28.1
-loguru==0.7.0
-matplotlib==3.7.1
-ninja==1.11.1
-opencv-python-headless==4.7.0.72
-pyrallis==0.3.1
-torch==2.1.1
-torchvision>=0.16
-tqdm==4.65.0
-transformers==4.29.1
-trimesh==3.21.6
-xatlas==0.0.7
+torch==2.2.1
+pytorch-lightning==2.2.0
+torch-geometric==2.5.0
+pytorch-cluster==1.6.3
+scikit-learn-intelex==2024.1.0
+numpy==1.26.4
+scikit-image==0.22.0
+scipy==1.12.0
+pandas==1.5.3
+openpyxl==3.1.2
+overrides==7.7.0
+pykdtree==1.3.11
+laspy[laszip,lazrs]==2.5.3
+pillow==10.2.0
+intel-openmp==2024.0.2
+tqdm==4.66.2
+pyglet==1.5.28
+rtree==1.2.0
+tensorboard==2.16.2
+trimesh==3.23.5
+pysdf==0.1.9
+jsonargparse[signatures]==4.27.5
+
 spaces==0.23.2
 gradio==4.19.1