fix bugs

Dockerfile

@@ -1,53 +0,0 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.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 torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --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

packages.txt

@@ -0,0 +1,12 @@
+libglfw3-dev
+libgles2-mesa-dev
+libgl1
+freeglut3-dev
+unzip
+ffmpeg 
+libsm6 
+libxext6
+libgl1-mesa-dri
+libegl1-mesa
+libgbm1
+build-essential

requirements.txt

@@ -0,0 +1,47 @@
+opencv-python
+numpy
+trimesh
+plyfile
+pyyaml
+scikit-image
+scikit-learn
+chumpy
+tensorboard
+kornia
+loguru
+pycocotools
+yacs
+lmdb
+fire
+setuptools
+einops
+tqdm
+ipython
+gym
+transformers==4.44.2
+OmegaConf
+matplotlib
+gradio==5.0.2
+diffusers
+rembg
+segment_anything
+jaxtyping
+imageio
+iopath
+timm
+open3d
+pyrender
+pytorch-lightning
+smplx==0.1.28
+chumpy @ git+https://github.com/zerchen/chumpy
+xtcocotools
+pandas
+hydra-core
+hydra-submitit-launcher
+hydra-colorlog
+pyrootutils
+rich
+webdataset
+ultralytics
+dill
+git+https://github.com/luca-medeiros/lang-segment-anything.git