update voxelize_cuda

Dockerfile

@@ -1,47 +1,50 @@
-FROM nvidia/cuda:11.3.1-devel-ubuntu20.04
+FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
-# Set up a new user named "user" with user ID 1000
-RUN useradd -m -u 1000 user
-# Switch to the "user" user
-USER user
-
-ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
-ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
-ENV FORCE_CUDA=1
-
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    build-essential \
-    curl \
-    git \
+    libgl1 \ 
+    freeglut3-dev \
+    unzip \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    libfontconfig1 \
     libegl1-mesa-dev \
     libgl1-mesa-dev \
     libgles2-mesa-dev \
     libglib2.0-0 \
-    libsm6 \
-    libxext6 \
     libxrender1 \
-    python-is-python3 \
+    libgbm1 \
+    build-essential \
+    libeigen3-dev \
     python3.8 \
     python3-pip \
-    wget \
-    libgl1 \
-    freeglut3-dev \
-    unzip \
-    ffmpeg \
-    libgl1-mesa-dri \
-    libegl1-mesa \
-    libgbm1 \
-    libturbojpeg \
+    nvidia-cuda-toolkit \
+    opencv-python-headless  \
     && rm -rf /var/lib/apt/lists/*
 
 
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+
+# Switch to the "user" user
+USER user
+
+FROM python:3.8
+
+ENV PYTHONUNBUFFERED=1
+
+ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
+ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
+ENV FORCE_CUDA=1
+
 ENV CUDA_HOME=/usr/local/cuda
-ENV PATH=${CUDA_HOME}/bin:/home/user/.local/bin:/usr/bin:${PATH}
+ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
 
+
 # Set home to the user's home directory
 ENV HOME=/home/user \
     PYTHONPATH=$HOME/app \
@@ -52,19 +55,13 @@ ENV HOME=/home/user \
 	GRADIO_THEME=huggingface \
 	SYSTEM=spaces
 
-FROM python:3.8
-
 RUN pip install --upgrade pip ninja
 RUN pip install setuptools==69.5.1
 RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
 
 RUN python -c "import torch; print(torch.version.cuda)"
 COPY requirements.txt /tmp
-RUN cd /tmp && pip install -r requirements.txt --user
-
-RUN git clone git+https://github.com/YuliangXiu/neural_voxelization_layer.git /home/user/neural_voxelization_layer
-WORKDIR /home/user/neural_voxelization_layer
-RUN  python setup.py install --user
+RUN cd /tmp && pip install -r requirements.txt
 
 RUN pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl
 RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl
@@ -74,5 +71,9 @@ WORKDIR $HOME/app
 
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
+RUN chmod 755 $HOME/app
+
+ENV TRANSFORMERS_CACHE=/tmp
+ENV MPLCONFIGDIR=/tmp
 
 CMD ["python", "app.py"]

app.py

@@ -4,9 +4,16 @@
 import glob
 import gradio as gr
 import numpy as np
+import os
+import subprocess
 
 from apps.infer import generate_model
 
+if os.getenv('SYSTEM') == 'spaces':
+    subprocess.run('pip install pyembree'.split())
+    subprocess.run(
+        'pip install git+https://github.com/YuliangXiu/neural_voxelization_layer.git'.split())
+
 # running
 
 description = '''

requirements.txt

@@ -16,7 +16,7 @@ shapely==1.7.1
 rtree==0.9.7
 pytorch_lightning==1.2.5
 PyMCubes
-opencv-python
+opencv-python-headless==4.8.1.78
 opencv_contrib_python
 scikit-learn
 protobuf==3.20.0
@@ -25,4 +25,5 @@ iopath
 fvcore
 chumpy
 open3d
+gradio
 git+https://github.com/YuliangXiu/rembg.git