Support CUDA 12.4, torch 2.5.1

Dockerfile

@@ -1,4 +1,4 @@
-ARG CUDA_VERSION=12.1.0-devel
+ARG CUDA_VERSION=12.4.0-devel
 
 FROM --platform=amd64 docker.io/nvidia/cuda:${CUDA_VERSION}-ubuntu22.04
 

README.md

@@ -18,7 +18,7 @@ This is an extensive node suite that enables ComfyUI to process 3D inputs (Mesh
 - [Pre-builds](https://github.com/MrForExample/Comfy3D_Pre_Builds) are available for:
   - Windows 10/11, Ubuntu 22.04
   - Python 3.10/3.11/3.12
-  - CUDA 12.1/11.8
+  - CUDA 12.4/12.1/11.8
   - torch 2.3.0+cu121/cu118, torchvision 0.18.0+cu121/cu118
 - [install.py](install.py) will download & install Pre-builds automatically according to your runtime environment, if it couldn't find corresponding Pre-builds, then [build script](_Pre_Builds/_Build_Scripts/auto_build_all.py) will start automatically, if automatic build doesn't work for you, then please check out [Semi-Automatic Build Guide](_Pre_Builds/README.md#build-required-packages-semi-automatically)
 - If you have any missing node in any open Comfy3D workflow, try simply click [Install Missing Custom Nodes](https://github.com/ltdrdata/ComfyUI-Manager?tab=readme-ov-file#support-of-missing-nodes-installation) in ComfyUI-Manager

_Pre_Builds/_Build_Scripts/build_config.yaml

@@ -1,6 +1,6 @@
 repo_id: "MrForExample/Comfy3D_Pre_Builds"
-supported_cuda_versions: ["12.1", "11.8"]
-cuda_version: "12.1"
+supported_cuda_versions: ["12.4", "12.1", "11.8"]
+cuda_version: "12.4"
 
 # Packages inside this file will be pip downloaded before actual build start
 build_requirements: "build_requirements.txt"
@@ -17,17 +17,18 @@ build_base_packages: ["torch", "torchvision"]
 # All packages will be automatically installed at beginning of install.py follwing orders from top to bottom 
 remote_packages:
   torch:
-    version: "2.4.0"
+    version: "2.5.1"
     url: "https://download.pytorch.org/whl/${cuda_version}"
   
   torchvision:
-    version: "0.19.0"
+    version: "0.20.1"
     url: "https://download.pytorch.org/whl/${cuda_version}"
 
-  torch-scatter:
-    url_option: "--find-links"
-    url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html"
-
   xformers:
-    version: "0.0.27.post2"
-    url: "https://download.pytorch.org/whl/${cuda_version}"
+    version: "0.0.28.post3"
+    url: "https://download.pytorch.org/whl/${cuda_version}"
+
+  # At the time of writing, torch-scatter is not available for PyTorch 2.5.0
+  #torch-scatter:
+  #  url_option: "--find-links"
+  #  url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html"

_Pre_Builds/_Build_Scripts/dependencies.txt

@@ -1,4 +1,5 @@
 https://github.com/ashawkey/diff-gaussian-rasterization.git
 https://github.com/NVlabs/nvdiffrast.git
 https://github.com/ashawkey/kiuikit.git
-https://github.com/facebookresearch/pytorch3d.git
+https://github.com/facebookresearch/pytorch3d.git
+https://github.com/rusty1s/pytorch_scatter.git