Update

Dockerfile

@@ -1,13 +1,16 @@
 # Use NVIDIA CUDA image that matches PyTorch's CUDA 12.4 compilation
 FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
 
-# Install Python 3.10 and dependencies with cache mounts
+# Install Python 3.13 and dependencies with cache mounts
 RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     --mount=type=cache,target=/var/lib/apt,sharing=locked \
     apt-get update && apt-get install -y \
-    python3.10 \
-    python3.10-venv \
-    python3.10-dev \
+    software-properties-common && \
+    add-apt-repository ppa:deadsnakes/ppa && \
+    apt-get update && apt-get install -y \
+    python3.13 \
+    python3.13-venv \
+    python3.13-dev \
     python3-pip \
     git \
     build-essential \
@@ -16,8 +19,8 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     wget
 
 # Create symlinks for python
-RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
-    ln -sf /usr/bin/python3.10 /usr/bin/python
+RUN ln -sf /usr/bin/python3.13 /usr/bin/python3 && \
+    ln -sf /usr/bin/python3.13 /usr/bin/python
 
 # Set CUDA environment variables for runtime
 ENV CUDA_HOME=/usr/local/cuda \
@@ -46,12 +49,12 @@ ENV HOME=/home/user \
 COPY --chown=user pyproject.docker.toml ./pyproject.toml
 COPY --chown=user uv.docker.lock ./uv.lock
 
-# Download pre-built wheels (built with same PyTorch version)
+# Download pre-built wheels (built with PyTorch 2.6.0 + CUDA 12.4)
 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" && \
-    wget -O wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl \
-    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl"
+    wget -O wheels/gsplat-0.1.0-cp313-cp313-linux_x86_64.whl \
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp313-cp313-linux_x86_64.whl" && \
+    wget -O wheels/fused_ssim-0.0.1+b4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl \
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl"
 
 # Install dependencies with exact versions from lock file
 RUN --mount=type=cache,target=/tmp/uv-cache,sharing=locked,uid=1000,gid=1000 \

gradio_app.py

@@ -543,7 +543,7 @@ def create_interface():
                         info="Name for this training run",
                     )
                     num_gaussians = gr.Slider(
-                        minimum=1000,
+                        minimum=100,
                         maximum=50000,
                         value=10000,
                         step=1000,

gradio_models.py

@@ -152,7 +152,10 @@ class GradioGaussianSplatting2D(nn.Module):
         self.gt_images = torch.from_numpy(self.gt_images).to(
             dtype=self.dtype, device=self.device
         )
-        self.img_h, self.img_w = self.gt_images.shape[1:]
+        self.img_h, self.img_w = (
+            int(self.gt_images.shape[1]),
+            int(self.gt_images.shape[2]),
+        )
         self.tile_bounds = (
             int((self.img_w + self.block_w - 1) // self.block_w),
             int((self.img_h + self.block_h - 1) // self.block_h),
@@ -394,13 +397,13 @@ class GradioGaussianSplatting2D(nn.Module):
         xy, radii, conics, num_tiles_hit = tmp
 
         # Ensure correct tensor types for CUDA backend
-        num_tiles_hit = num_tiles_hit.to(dtype=torch.int32)  # Must be int32
-        radii = radii.to(dtype=torch.float32)  # Must be float32
+        # Note: The custom gsplat CUDA code expects int32 for both radii and num_tiles_hit
+        num_tiles_hit = num_tiles_hit.to(dtype=torch.int32)
+        radii = radii.to(dtype=torch.int32)
 
         if not self.disable_tiles:
             enable_topk_norm = not self.disable_topk_norm
 
-
             out_image = rasterize_gaussians_sum(
                 xy,
                 radii,
@@ -481,8 +484,9 @@ class GradioGaussianSplatting2D(nn.Module):
             xy, radii, conics, num_tiles_hit = tmp
 
             # Ensure correct tensor types for CUDA backend
-            num_tiles_hit = num_tiles_hit.to(dtype=torch.int32)  # Must be int32
-            radii = radii.to(dtype=torch.float32)  # Must be float32
+            # Note: The custom gsplat CUDA code expects int32 for both radii and num_tiles_hit
+            num_tiles_hit = num_tiles_hit.to(dtype=torch.int32)
+            radii = radii.to(dtype=torch.int32)
 
             if not self.disable_tiles:
                 enable_topk_norm = not self.disable_topk_norm

pyproject.docker.toml

@@ -3,7 +3,7 @@ name = "image-gs"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.13"
 dependencies = [
     "lpips>=0.1.4",
     "matplotlib>=3.10.6",
@@ -21,16 +21,12 @@ dependencies = [
     "huggingface_hub>=0.24.0",
 ]
 
-# We use python 3.10 and cu124
+# We use python 3.13 and cu124 with PyTorch 2.6.0
 [tool.uv.sources]
-fused_ssim = { path = "wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl" }
-gsplat = { path = "wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" }
-torch = [
-    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
-]
-torchvision = [
-    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
-]
+fused_ssim = { path = "wheels/fused_ssim-0.0.1+b4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl" }
+gsplat = { path = "wheels/gsplat-0.1.0-cp313-cp313-linux_x86_64.whl" }
+torch = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]
+torchvision = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]
 
 
 [tool.uv.extra-build-dependencies]
@@ -42,7 +38,4 @@ url = "https://download.pytorch.org/whl/cu124"
 explicit = true
 
 [dependency-groups]
-dev = [
-    "huggingface-hub[cli]>=0.34.4",
-]
-
+dev = ["huggingface-hub[cli]>=0.34.4"]

pyproject.toml

@@ -23,18 +23,14 @@ dependencies = [
 
 # We use python 3.10 and cu124
 [tool.uv.sources]
-fused_ssim = { git = "https://github.com/rahul-goel/fused-ssim/" }
+fused_ssim = { url = "https://github.com/MiroPsota/torch_packages_builder/releases/download/fused_ssim-0.0.1%2Bb4fd832/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl" }
 gsplat = { path = "gsplat" }
-torch = [
-    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
-]
-torchvision = [
-    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
-]
+torch = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]
+torchvision = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]
 
 
 [tool.uv.extra-build-dependencies]
-fused-ssim = ["torch", "numpy"]
+# fused-ssim = ["torch", "numpy"]  # Not needed - using pre-built wheel
 gsplat = ["torch", "numpy"]
 
 [[tool.uv.index]]
@@ -43,6 +39,4 @@ url = "https://download.pytorch.org/whl/cu124"
 explicit = true
 
 [dependency-groups]
-dev = [
-    "huggingface-hub[cli]>=0.34.4",
-]
+dev = ["huggingface-hub[cli]>=0.34.4"]

uv.lock

@@ -547,8 +547,11 @@ wheels = [
 
 [[package]]
 name = "fused-ssim"
-version = "0.0.0"
-source = { git = "https://github.com/rahul-goel/fused-ssim/#b4fd8324e81c48c9b2b9f62e1b9c6431fece6ab3" }
+version = "0.0.1+b4fd832pt2.6.0cu124"
+source = { url = "https://github.com/MiroPsota/torch_packages_builder/releases/download/fused_ssim-0.0.1%2Bb4fd832/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl" }
+wheels = [
+    { url = "https://github.com/MiroPsota/torch_packages_builder/releases/download/fused_ssim-0.0.1%2Bb4fd832/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl", hash = "sha256:51c2b3302406ba972ab79dd31b3487eb282f1ba18a15e68ab11ea77b95bb0868" },
+]
 
 [[package]]
 name = "gradio"
@@ -770,7 +773,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "fused-ssim", git = "https://github.com/rahul-goel/fused-ssim/" },
+    { name = "fused-ssim", url = "https://github.com/MiroPsota/torch_packages_builder/releases/download/fused_ssim-0.0.1%2Bb4fd832/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl" },
     { name = "gradio", specifier = ">=4.0.0" },
     { name = "gsplat", directory = "gsplat" },
     { name = "huggingface-hub", specifier = ">=0.24.0" },