Fix PyTorch version compatibility for gsplat wheel

- Pin PyTorch to exact version 2.6.0 and torchvision to 0.21.0 in Docker config
- Copy uv.lock to ensure exact dependency versions match local environment
- Use --frozen flag to prevent version drift during Docker build
- Rebuild and upload gsplat wheel with matching PyTorch 2.6.0+cu124
- This fixes undefined symbol errors from PyTorch version mismatches

Dockerfile

@@ -42,18 +42,18 @@ ENV HOME=/home/user \
     GRADIO_SERVER_PORT=7860 \
     UV_CACHE_DIR=/home/user/.cache/uv
 
-# Copy dependency files
-COPY --chown=user pyproject.toml ./pyproject.toml
-COPY --chown=user gsplat/ ./gsplat/
+# Copy dependency files with exact versions
+COPY --chown=user pyproject.docker.toml ./pyproject.toml
+COPY --chown=user uv.lock ./uv.lock
 
-# Set minimal CUDA compilation to avoid memory issues but ensure compatibility
-ENV TORCH_CUDA_ARCH_LIST="8.0" \
-    FORCE_CUDA=1 \
-    MAX_JOBS=1
+# Download optimized gsplat wheel (built with same PyTorch version)
+RUN mkdir -p wheels && \
+    wget -O wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl \
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl"
 
-# Install dependencies (will compile gsplat with exact PyTorch version for compatibility)
+# Install dependencies with exact versions from lock file
 RUN --mount=type=cache,target=/tmp/uv-cache,sharing=locked,uid=1000,gid=1000 \
-    UV_CACHE_DIR=/tmp/uv-cache uv sync --no-dev
+    UV_CACHE_DIR=/tmp/uv-cache uv sync --frozen --no-dev
 
 # Copy the rest of the application
 COPY --chown=user . .

pyproject.docker.toml

@@ -11,9 +11,9 @@ dependencies = [
     "pytorch-msssim>=1.0.0",
     "scikit-image>=0.25.2",
     "scipy>=1.15.3",
-    "torch>=2.6.0",
+    "torch==2.6.0",
     "torchmetrics>=1.8.2",
-    "torchvision>=0.21.0",
+    "torchvision==0.21.0",
     "fused_ssim",
     "pyyaml>=6.0.2",
     "gsplat",
@@ -24,7 +24,7 @@ dependencies = [
 # We use python 3.10 and cu124
 [tool.uv.sources]
 fused_ssim = { git = "https://github.com/rahul-goel/fused-ssim/" }
-gsplat = { path = "gsplat" }
+gsplat = { path = "wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" }
 torch = [
     { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
 ]