Match the CUDA version (12.1) to ZeroGPU.

build-cuda-ext.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+#
+# @File:   build-cuda-ext.sh
+# @Author: Haozhe Xie
+# @Date:   2024-09-22 10:43:34
+# @Last Modified by: Haozhe Xie
+# @Last Modified at: 2024-09-22 10:46:29
+# @Email:  root@haozhexie.com
+
+SCRIPT_DIR=$(dirname "$(readlink -f "$0")")
+EXTENSIONS_DIR="$SCRIPT_DIR/citydreamer/extensions"
+WHEELS_DIR="$SCRIPT_DIR/wheels"
+
+mkdir -p "$WHEELS_DIR"
+
+for dir in "$EXTENSIONS_DIR"/*/; do
+  if [ -d "$dir" ]; then
+    echo "Processing $dir"
+    (cd "$dir" && pip wheel .)
+    mv "$dir"*.whl "$WHEELS_DIR"
+  fi
+done

requirements.txt

@@ -1,5 +1,5 @@
-torch==2.2.0
-torchvision==0.17.0
+torch==2.2.2
+torchvision==0.17.2
 
 easydict
 gradio

wheels/extrude_tensor-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:58b87e506ae559cb11e6cb1375bdc6c119f02f3d9560b015b46d3561048b3fd2
-size 237463
+oid sha256:2c7e143d972796e4092b874de35bcca064885539352cfd9d3c652ad3a7a9ca40
+size 237842

wheels/grid_encoder-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b271befcb1ed3603cf0d62d0599ad7288a70886c98a66b350fdf4a64529ed3f
-size 2375579
+oid sha256:7de95ace4abdb055a0326538854eb1ce173440ac2b0be5fce1d4cfa81522ebc1
+size 2345063

wheels/voxrender-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e65240d2379e48f9e4a541b146ad9b3f66bc0b6549c0732bfc2aa63bb7abc7eb
-size 593162
+oid sha256:e56e7d57812d7778339e64b8bc2e14ea52eb37dead72c290650a163371a0cd4c
+size 593569