switch to docker

Dockerfile

@@ -0,0 +1,61 @@
+FROM nvidia/cuda:11.8.0-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 setuptools ninja
+RUN pip install torch==2.2.1 --extra-index-url https://download.pytorch.org/whl/cu118
+
+RUN python -c "import torch; print(torch.version.cuda)"
+COPY requirements.txt /tmp
+RUN cd /tmp && pip install -r requirements.txt
+
+# 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

@@ -1,10 +1,10 @@
 ---
 title: TripoSR
-emoji: 🏃
+emoji: 🐳
 colorFrom: gray
 colorTo: red
-sdk: gradio
-sdk_version: 4.19.2
+sdk: docker
+# sdk_version: 4.19.2
 app_file: app.py
 pinned: false
 license: mit

pre-requirements.txt

@@ -1 +0,0 @@
-torch==2.1.2

requirements.txt

@@ -6,3 +6,4 @@ transformers==4.35.0
 trimesh==4.0.5
 rembg
 huggingface-hub
+gradio