use docker for cuda-11.3

Dockerfile

@@ -0,0 +1,78 @@
+FROM nvidia/cuda:11.3.1-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 \
+    libegl1-mesa-dev \
+    libgl1-mesa-dev \
+    libgles2-mesa-dev \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender1 \
+    python-is-python3 \
+    python3.8 \
+    python3-pip \
+    wget \
+    libgl1 \
+    freeglut3-dev \
+    unzip \
+    ffmpeg \
+    libgl1-mesa-dri \
+    libegl1-mesa \
+    libgbm1 \
+    libturbojpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+
+ENV CUDA_HOME=/usr/local/cuda
+ENV PATH=${CUDA_HOME}/bin:/home/user/.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 \
+	PYTHONUNBUFFERED=1 \
+	GRADIO_ALLOW_FLAGGING=never \
+	GRADIO_NUM_PORTS=1 \
+	GRADIO_SERVER_NAME=0.0.0.0 \
+	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 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
+
+# Set the working directory to the user's home directory
+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
+
+CMD ["python", "app.py"]

README.md

@@ -4,11 +4,9 @@ metaTitle: Making yourself an ICON, by Yuliang Xiu
 emoji: 🤼
 colorFrom: indigo
 colorTo: yellow
-sdk: gradio
-sdk_version: 4.36.1
+sdk: docker
 app_file: app.py
 pinned: true
-python_version: 3.8.13
 ---
 
 # ICON Clothed Human Digitization 

app.py

@@ -3,22 +3,8 @@
 
 import glob
 import gradio as gr
-import os
 import numpy as np
 
-import subprocess
-
-if os.getenv('SYSTEM') == 'spaces':
-    subprocess.run('pip install pyembree'.split())
-    subprocess.run(
-        'pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html'.split())
-    subprocess.run(
-        'pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl'.split())
-    subprocess.run(
-        'pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl'.split())
-    subprocess.run(
-        'pip install git+https://github.com/YuliangXiu/neural_voxelization_layer.git'.split())
-
 from apps.infer import generate_model
 
 # running

packages.txt

@@ -1,11 +0,0 @@
-libgl1
-freeglut3-dev
-unzip
-ffmpeg 
-libsm6 
-libxext6
-libgl1-mesa-dri
-libegl1-mesa
-libgbm1
-build-essential
-libturbojpeg

requirements.txt

@@ -25,4 +25,4 @@ iopath
 fvcore
 chumpy
 open3d
-git+https://github.com/YuliangXiu/rembg.git
+git+https://github.com/YuliangXiu/rembg.git