set up docker sdk

Dockerfile

@@ -0,0 +1,66 @@
+FROM nvidia/cuda:11.3.1-devel-ubuntu20.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+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_NAME}/.local/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
+
+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-dev \
+    python3-pip \
+    wget \
+    && 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
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH \
+    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
+
+RUN pip install --upgrade pip ninja
+RUN pip install setuptools==69.5.1
+RUN pip install torch==1.10+cu113 torchvision==0.11.0+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
+
+RUN python -c "import torch; print(torch.version.cuda)"
+COPY requirements.txt /tmp
+RUN cd /tmp && pip install -r requirements.txt
+
+RUN git clone -b v0.7.4 https://github.com/facebookresearch/pytorch3d.git /home/user/pytorch3d-0.7.4
+WORKDIR /home/user/pytorch3d-0.7.4
+RUN  python setup.py install --user
+
+# 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

@@ -3,9 +3,7 @@ title: 3DFauna Demo
 emoji: 👀
 colorFrom: blue
 colorTo: green
-sdk: gradio
-python_version: 3.9.13
-sdk_version: 3.50.2
+sdk: docker
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-4.0

app.py

@@ -8,9 +8,9 @@ import sys
 import torch
 
 if os.getenv('SYSTEM') == 'spaces':
-    os.system('pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
-    os.system('pip install fvcore iopath')
-    os.system('pip install git+https://github.com/facebookresearch/pytorch3d.git')
+    os.system('pip install --global-option="--no-networks" git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
+    # os.system('pip install fvcore iopath')
+    # os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git"')
     
 import cv2
 import time

requirements-old.txt

@@ -0,0 +1,30 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch==1.10+cu113
+torchvision==0.11.0+cu113
+fvcore
+iopath
+ninja
+ConfigArgParse==1.5.3
+einops==0.4.1
+fire==0.5.0
+glfw==2.5.7
+gradio==4.12.0
+imageio==2.27.0
+ipdb==0.13.9
+lpips==0.1.4
+matplotlib==3.8.1
+numpy==1.23.1
+Pillow==9.2.0
+PyOpenGL==3.1.6
+PyYAML==6.0
+scipy==1.9.1
+segment_anything==1.0
+siren_pytorch==0.1.7
+transformers==4.28.1
+trimesh==4.0.0
+wandb==0.14.2
+xatlas==0.0.7
+tensorboard==2.10.0
+git+https://github.com/NVlabs/nvdiffrast/
+opencv-python
+opencv_contrib_python

requirements.txt

@@ -1,6 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.10+cu113
-torchvision==0.11.0+cu113
 fvcore
 iopath
 ninja