cpu fallback

Dockerfile

@@ -6,12 +6,14 @@ RUN wget -q https://developer.download.nvidia.com/compute/cuda/11.6.0/local_inst
 RUN apt-get update && apt-get install -y dkms build-essential gcc g++ libxi6 libxi-dev libglu1-mesa libglu1-mesa-dev libxmu6 linux-headers-amd64 linux-source freeglut3-dev
 RUN sh cuda_11.6.0_510.39.01_linux.run --silent --toolkit
 WORKDIR /home/user/app
-#RUN --mount=target=/root/packages.txt,source=packages.txt 	apt-get update && 	xargs -r -a /root/packages.txt apt-get install -y 	&& rm -rf /var/lib/apt/lists/*
-#Restoring cache
 RUN --mount=target=pre-requirements.txt,source=pre-requirements.txt 	pip install --no-cache-dir -r pre-requirements.txt
 ENV TORCH_CUDA_ARCH_LIST=Turing
 RUN --mount=target=requirements1.txt,source=requirements1.txt 	pip install --no-cache-dir -r requirements1.txt
 RUN --mount=target=requirements2.txt,source=requirements2.txt 	pip install --no-cache-dir -r requirements2.txt
 RUN pip install gradio
-COPY --chown=user . /home/usr/app
-ENTRYPOINT ["python", "/home/usr/app/app.py"]
+COPY --chown=user . /home/user/app
+RUN wget https://huggingface.co/camenduru/Dynamic3DGaussians/resolve/main/output.zip?download=true && unzip output.zip && rm output.zip
+RUN cp -R /home/user/app/Dynamic3DGaussians /home/user/app/Dynamic3DGaussians_cpu
+RUN sed -i s/\.cuda\(\)\./\./ /home/user/app/Dynamic3DGaussians_cpu/*.py
+RUN sed -i s/, device="cuda"// /home/user/app/Dynamic3DGaussians_cpu/*.py
+ENTRYPOINT ["python", "/home/user/app/app.py"]

app.py

@@ -1,8 +1,13 @@
+import torch
 import sys
-sys.path.append("/home/usr/app/Dynamic3DGaussians")
+if torch.cuda.is_available():
+	sys.path.append("/home/user/app/Dynamic3DGaussians")
+	from Dynamic3DGaussians.visualize import visualize
+else:
+	sys.path.append("/home/user/app/Dynamic3DGaussians_cpu")
+	from Dynamic3DGaussians_cpu.visualize import visualize
 
 import gradio as gr
-from Dynamic3DGaussians.visualize import visualize
 
 def basketball(name):
     visualize("basketball", "pretrained")

requirements.txt

@@ -1,7 +0,0 @@
-pip==22.3.1
-torchaudio==0.12.1
-torchvision==0.13.1
-tqdm==4.65.0
-Pillow==9.4.0
-open3d==0.16.0
-diff-gaussian-rasterization @ git+https://github.com/JonathonLuiten/diff-gaussian-rasterization-w-depth.git@cb65e4b86bc3bd8ed42174b72a62e8d3a3a71110